Avoided transactional truncates on amateur databases.

Fixed a test failure that appeared after 753a22a6, although the bug
existed before that commit.

Refs #22308.
This commit is contained in:
Aymeric Augustin 2014-03-22 14:36:01 +01:00
parent ee4edb1eda
commit 6877a9d415
1 changed files with 5 additions and 4 deletions

View File

@ -28,8 +28,8 @@ class Command(NoArgsCommand):
're-executed, and the initial_data fixture will be re-installed.')
def handle_noargs(self, **options):
db = options.get('database')
connection = connections[db]
database = options.get('database')
connection = connections[database]
verbosity = int(options.get('verbosity'))
interactive = options.get('interactive')
# The following are stealth options used by Django's internals.
@ -63,7 +63,8 @@ Are you sure you want to do this?
if confirm == 'yes':
try:
with transaction.atomic(using=db):
with transaction.atomic(using=database,
savepoint=connection.features.can_rollback_ddl):
with connection.cursor() as cursor:
for sql in sql_list:
cursor.execute(sql)
@ -78,7 +79,7 @@ Are you sure you want to do this?
six.reraise(CommandError, CommandError(new_msg), sys.exc_info()[2])
if not inhibit_post_migrate:
self.emit_post_migrate(verbosity, interactive, db)
self.emit_post_migrate(verbosity, interactive, database)
# Reinstall the initial_data fixture.
if options.get('load_initial_data'):