diff options
Diffstat (limited to 'django/db/backends/schema.py')
| -rw-r--r-- | django/db/backends/schema.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/django/db/backends/schema.py b/django/db/backends/schema.py index d187c48046..fe28be92bc 100644 --- a/django/db/backends/schema.py +++ b/django/db/backends/schema.py @@ -72,7 +72,8 @@ class BaseDatabaseSchemaEditor(object): def __enter__(self): self.deferred_sql = [] if self.connection.features.can_rollback_ddl: - atomic(self.connection.alias).__enter__() + self.atomic = atomic(self.connection.alias) + self.atomic.__enter__() return self def __exit__(self, exc_type, exc_value, traceback): @@ -80,7 +81,7 @@ class BaseDatabaseSchemaEditor(object): for sql in self.deferred_sql: self.execute(sql) if self.connection.features.can_rollback_ddl: - atomic(self.connection.alias).__exit__(exc_type, exc_value, traceback) + self.atomic.__exit__(exc_type, exc_value, traceback) # Core utility functions |