Refs #33308 -- Added psycopg_any.IsolationLevel.

2 files changed, 24 insertions, 6 deletions

diff --git a/django/db/backends/postgresql/base.py b/django/db/backends/postgresql/base.py
index 995510bcb2..0aee39aa5c 100644
--- a/django/db/backends/postgresql/base.py
+++ b/django/db/backends/postgresql/base.py
@@ -48,6 +48,7 @@ from .creation import DatabaseCreation  # NOQA
 from .features import DatabaseFeatures  # NOQA
 from .introspection import DatabaseIntrospection  # NOQA
 from .operations import DatabaseOperations  # NOQA
+from .psycopg_any import IsolationLevel  # NOQA
 from .schema import DatabaseSchemaEditor  # NOQA
 
 psycopg2.extensions.register_adapter(SafeString, psycopg2.extensions.QuotedString)
@@ -212,22 +213,30 @@ class DatabaseWrapper(BaseDatabaseWrapper):
 
     @async_unsafe
     def get_new_connection(self, conn_params):
-        connection = Database.connect(**conn_params)
-
         # self.isolation_level must be set:
         # - after connecting to the database in order to obtain the database's
         #   default when no value is explicitly specified in options.
         # - before calling _set_autocommit() because if autocommit is on, that
         #   will set connection.isolation_level to ISOLATION_LEVEL_AUTOCOMMIT.
         options = self.settings_dict["OPTIONS"]
+        set_isolation_level = False
         try:
-            self.isolation_level = options["isolation_level"]
+            isolation_level_value = options["isolation_level"]
         except KeyError:
-            self.isolation_level = connection.isolation_level
+            self.isolation_level = IsolationLevel.READ_COMMITTED
         else:
             # Set the isolation level to the value from OPTIONS.
-            if self.isolation_level != connection.isolation_level:
-                connection.set_session(isolation_level=self.isolation_level)
+            try:
+                self.isolation_level = IsolationLevel(isolation_level_value)
+                set_isolation_level = True
+            except ValueError:
+                raise ImproperlyConfigured(
+                    f"Invalid transaction isolation level {isolation_level_value} "
+                    f"specified. Use one of the IsolationLevel values."
+                )
+        connection = Database.connect(**conn_params)
+        if set_isolation_level:
+            connection.isolation_level = self.isolation_level
         # Register dummy loads() to avoid a round trip from psycopg2's decode
         # to json.dumps() to json.loads(), when using a custom decoder in
         # JSONField.
diff --git a/django/db/backends/postgresql/psycopg_any.py b/django/db/backends/postgresql/psycopg_any.py
index 8e0d170867..83e8a9f4d3 100644
--- a/django/db/backends/postgresql/psycopg_any.py
+++ b/django/db/backends/postgresql/psycopg_any.py
@@ -1,3 +1,5 @@
+from enum import IntEnum
+
 from psycopg2 import errors, extensions, sql  # NOQA
 from psycopg2.extras import DateRange, DateTimeRange, DateTimeTZRange, Inet  # NOQA
 from psycopg2.extras import Json as Jsonb  # NOQA
@@ -6,6 +8,13 @@ from psycopg2.extras import NumericRange, Range  # NOQA
 RANGE_TYPES = (DateRange, DateTimeRange, DateTimeTZRange, NumericRange)
 
 
+class IsolationLevel(IntEnum):
+    READ_UNCOMMITTED = extensions.ISOLATION_LEVEL_READ_UNCOMMITTED
+    READ_COMMITTED = extensions.ISOLATION_LEVEL_READ_COMMITTED
+    REPEATABLE_READ = extensions.ISOLATION_LEVEL_REPEATABLE_READ
+    SERIALIZABLE = extensions.ISOLATION_LEVEL_SERIALIZABLE
+
+
 def _quote(value, connection=None):
     adapted = extensions.adapt(value)
     if hasattr(adapted, "encoding"):