Commit 3becac84 authored by Aymeric Augustin's avatar Aymeric Augustin
Browse files

Fixed #22321 -- Wrapped exceptions in _set_autocommit.

Refs #21202.
parent d5031ecf
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -488,6 +488,7 @@ class DatabaseWrapper(BaseDatabaseWrapper):
            pass

    def _set_autocommit(self, autocommit):
        with self.wrap_database_errors:
            self.connection.autocommit(autocommit)

    def disable_constraint_checking(self):
+2 −1
Original line number Diff line number Diff line
@@ -687,6 +687,7 @@ class DatabaseWrapper(BaseDatabaseWrapper):
        pass

    def _set_autocommit(self, autocommit):
        with self.wrap_database_errors:
            self.connection.autocommit = autocommit

    def check_constraints(self, table_names=None):
+8 −7
Original line number Diff line number Diff line
@@ -189,6 +189,7 @@ class DatabaseWrapper(BaseDatabaseWrapper):
            self.connection.set_isolation_level(isolation_level)

    def _set_autocommit(self, autocommit):
        with self.wrap_database_errors:
            if self.psycopg2_version >= (2, 4, 2):
                self.connection.autocommit = autocommit
            else:
+2 −1
Original line number Diff line number Diff line
@@ -411,6 +411,7 @@ class DatabaseWrapper(BaseDatabaseWrapper):
            level = ''
        # 'isolation_level' is a misleading API.
        # SQLite always runs at the SERIALIZABLE isolation level.
        with self.wrap_database_errors:
            self.connection.isolation_level = level

    def check_constraints(self, table_names=None):