Fixed #31516 -- Improved naming of migrations with multiple operations.

52 gives 60 in total (52 + 5 + 3).

Co-authored-by: Adam Johnson <me@adamj.eu>
This commit is contained in:
manav014 2021-03-10 19:04:44 +05:30 committed by Mariusz Felisiak
parent 927fead2f5
commit 99640e24b4
2 changed files with 57 additions and 12 deletions

View File

@ -1,4 +1,3 @@
from django.db.migrations import operations
from django.db.migrations.utils import get_migration_name_timestamp from django.db.migrations.utils import get_migration_name_timestamp
from django.db.transaction import atomic from django.db.transaction import atomic
@ -186,16 +185,19 @@ class Migration:
if self.initial: if self.initial:
return 'initial' return 'initial'
name = None raw_fragments = [op.migration_name_fragment for op in self.operations]
if len(self.operations) == 1: fragments = [name for name in raw_fragments if name]
name = self.operations[0].migration_name_fragment
elif ( if not fragments or len(fragments) != len(self.operations):
len(self.operations) > 1 and return 'auto_%s' % get_migration_name_timestamp()
all(isinstance(o, operations.CreateModel) for o in self.operations)
): name = fragments[0]
name = '_'.join(sorted(o.migration_name_fragment for o in self.operations)) for fragment in fragments[1:]:
if name is None: new_name = f'{name}_{fragment}'
name = 'auto_%s' % get_migration_name_timestamp() if len(new_name) > 52:
name = f'{name}_and_more'
break
name = new_name
return name return name

View File

@ -2661,6 +2661,23 @@ class MigrationSuggestNameTests(SimpleTestCase):
migration = Migration('0002_initial', 'test_app') migration = Migration('0002_initial', 'test_app')
self.assertEqual(migration.suggest_name(), 'delete_person') self.assertEqual(migration.suggest_name(), 'delete_person')
def test_single_operation_long_name(self):
class Migration(migrations.Migration):
operations = [migrations.CreateModel('A' * 53, fields=[])]
migration = Migration('some_migration', 'test_app')
self.assertEqual(migration.suggest_name(), 'a' * 53)
def test_two_operations(self):
class Migration(migrations.Migration):
operations = [
migrations.CreateModel('Person', fields=[]),
migrations.DeleteModel('Animal'),
]
migration = Migration('some_migration', 'test_app')
self.assertEqual(migration.suggest_name(), 'person_delete_animal')
def test_two_create_models(self): def test_two_create_models(self):
class Migration(migrations.Migration): class Migration(migrations.Migration):
operations = [ operations = [
@ -2669,7 +2686,7 @@ class MigrationSuggestNameTests(SimpleTestCase):
] ]
migration = Migration('0001_initial', 'test_app') migration = Migration('0001_initial', 'test_app')
self.assertEqual(migration.suggest_name(), 'animal_person') self.assertEqual(migration.suggest_name(), 'person_animal')
def test_two_create_models_with_initial_true(self): def test_two_create_models_with_initial_true(self):
class Migration(migrations.Migration): class Migration(migrations.Migration):
@ -2682,6 +2699,32 @@ class MigrationSuggestNameTests(SimpleTestCase):
migration = Migration('0001_initial', 'test_app') migration = Migration('0001_initial', 'test_app')
self.assertEqual(migration.suggest_name(), 'initial') self.assertEqual(migration.suggest_name(), 'initial')
def test_many_operations_suffix(self):
class Migration(migrations.Migration):
operations = [
migrations.CreateModel('Person1', fields=[]),
migrations.CreateModel('Person2', fields=[]),
migrations.CreateModel('Person3', fields=[]),
migrations.DeleteModel('Person4'),
migrations.DeleteModel('Person5'),
]
migration = Migration('some_migration', 'test_app')
self.assertEqual(
migration.suggest_name(),
'person1_person2_person3_delete_person4_and_more',
)
def test_operation_with_no_suggested_name(self):
class Migration(migrations.Migration):
operations = [
migrations.CreateModel('Person', fields=[]),
migrations.RunSQL('SELECT 1 FROM person;'),
]
migration = Migration('some_migration', 'test_app')
self.assertIs(migration.suggest_name().startswith('auto_'), True)
def test_none_name(self): def test_none_name(self):
class Migration(migrations.Migration): class Migration(migrations.Migration):
operations = [migrations.RunSQL('SELECT 1 FROM person;')] operations = [migrations.RunSQL('SELECT 1 FROM person;')]