Fixed #31503 -- Made autodetector remove unique/index_together before altering fields.

This commit is contained in:
David Wobrock 2021-10-16 11:20:33 +02:00 committed by Mariusz Felisiak
parent c1e4111c74
commit 0314593fe8
2 changed files with 181 additions and 19 deletions

View File

@ -178,8 +178,12 @@ class MigrationAutodetector:
# Generate index removal operations before field is removed # Generate index removal operations before field is removed
self.generate_removed_constraints() self.generate_removed_constraints()
self.generate_removed_indexes() self.generate_removed_indexes()
# Generate field operations # Generate field renaming operations.
self.generate_renamed_fields() self.generate_renamed_fields()
# Generate removal of foo together.
self.generate_removed_altered_unique_together()
self.generate_removed_altered_index_together()
# Generate field operations.
self.generate_removed_fields() self.generate_removed_fields()
self.generate_added_fields() self.generate_added_fields()
self.generate_altered_fields() self.generate_altered_fields()
@ -1128,8 +1132,7 @@ class MigrationAutodetector:
dependencies.append((through_app_label, through_object_name, None, True)) dependencies.append((through_app_label, through_object_name, None, True))
return dependencies return dependencies
def _generate_altered_foo_together(self, operation): def _get_altered_foo_together_operations(self, option_name):
option_name = operation.option_name
for app_label, model_name in sorted(self.kept_model_keys): for app_label, model_name in sorted(self.kept_model_keys):
old_model_name = self.renamed_models.get((app_label, model_name), model_name) old_model_name = self.renamed_models.get((app_label, model_name), model_name)
old_model_state = self.from_state.models[app_label, old_model_name] old_model_state = self.from_state.models[app_label, old_model_name]
@ -1157,13 +1160,49 @@ class MigrationAutodetector:
dependencies.extend(self._get_dependencies_for_foreign_key( dependencies.extend(self._get_dependencies_for_foreign_key(
app_label, model_name, field, self.to_state, app_label, model_name, field, self.to_state,
)) ))
yield (
old_value,
new_value,
app_label,
model_name,
dependencies,
)
def _generate_removed_altered_foo_together(self, operation):
for (
old_value,
new_value,
app_label,
model_name,
dependencies,
) in self._get_altered_foo_together_operations(operation.option_name):
removal_value = new_value.intersection(old_value)
if removal_value or old_value:
self.add_operation( self.add_operation(
app_label, app_label,
operation( operation(name=model_name, **{operation.option_name: removal_value}),
name=model_name, dependencies=dependencies,
**{option_name: new_value} )
),
def generate_removed_altered_unique_together(self):
self._generate_removed_altered_foo_together(operations.AlterUniqueTogether)
def generate_removed_altered_index_together(self):
self._generate_removed_altered_foo_together(operations.AlterIndexTogether)
def _generate_altered_foo_together(self, operation):
for (
old_value,
new_value,
app_label,
model_name,
dependencies,
) in self._get_altered_foo_together_operations(operation.option_name):
removal_value = new_value.intersection(old_value)
if new_value != removal_value:
self.add_operation(
app_label,
operation(name=model_name, **{operation.option_name: new_value}),
dependencies=dependencies, dependencies=dependencies,
) )

View File

@ -1570,9 +1570,26 @@ class AutodetectorTests(TestCase):
) )
# Right number/type of migrations? # Right number/type of migrations?
self.assertNumberMigrations(changes, "otherapp", 1) self.assertNumberMigrations(changes, "otherapp", 1)
self.assertOperationTypes(changes, "otherapp", 0, ["AlterUniqueTogether", "AlterIndexTogether"]) self.assertOperationTypes(changes, 'otherapp', 0, [
self.assertOperationAttributes(changes, "otherapp", 0, 0, name="book", unique_together={("title", "author")}) 'AlterUniqueTogether',
self.assertOperationAttributes(changes, "otherapp", 0, 1, name="book", index_together={("title", "author")}) 'AlterIndexTogether',
'AlterUniqueTogether',
'AlterIndexTogether',
])
self.assertOperationAttributes(
changes, 'otherapp', 0, 0, name='book', unique_together=set(),
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 1, name='book', index_together=set(),
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 2, name='book',
unique_together={('title', 'author')},
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 3, name='book',
index_together={('title', 'author')},
)
def test_add_field_and_foo_together(self): def test_add_field_and_foo_together(self):
""" """
@ -1619,10 +1636,100 @@ class AutodetectorTests(TestCase):
) )
# Right number/type of migrations? # Right number/type of migrations?
self.assertNumberMigrations(changes, "otherapp", 1) self.assertNumberMigrations(changes, "otherapp", 1)
self.assertOperationTypes(changes, "otherapp", 0, ["AlterUniqueTogether", "AlterIndexTogether", "RemoveField"]) self.assertOperationTypes(changes, 'otherapp', 0, [
self.assertOperationAttributes(changes, "otherapp", 0, 0, name="book", unique_together={("author", "title")}) 'AlterUniqueTogether',
self.assertOperationAttributes(changes, "otherapp", 0, 1, name="book", index_together={("author", "title")}) 'AlterIndexTogether',
self.assertOperationAttributes(changes, "otherapp", 0, 2, model_name="book", name="newfield") 'AlterUniqueTogether',
'AlterIndexTogether',
'RemoveField',
])
self.assertOperationAttributes(
changes, 'otherapp', 0, 0, name='book', unique_together=set(),
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 1, name='book', index_together=set(),
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 2, name='book',
unique_together={('author', 'title')},
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 3, name='book',
index_together={('author', 'title')},
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 4, model_name='book', name='newfield',
)
def test_alter_field_and_foo_together(self):
"""Fields are altered after deleting some index/unique_together."""
initial_author = ModelState('testapp', 'Author', [
('id', models.AutoField(primary_key=True)),
('name', models.CharField(max_length=200)),
('age', models.IntegerField(db_index=True)),
], {
'unique_together': {('name',)},
})
author_reversed_constraints = ModelState('testapp', 'Author', [
('id', models.AutoField(primary_key=True)),
('name', models.CharField(max_length=200, unique=True)),
('age', models.IntegerField()),
], {
'index_together': {('age',)},
})
changes = self.get_changes([initial_author], [author_reversed_constraints])
self.assertNumberMigrations(changes, 'testapp', 1)
self.assertOperationTypes(changes, 'testapp', 0, [
'AlterUniqueTogether',
'AlterField',
'AlterField',
'AlterIndexTogether',
])
self.assertOperationAttributes(
changes, 'testapp', 0, 0, name='author', unique_together=set(),
)
self.assertOperationAttributes(
changes, 'testapp', 0, 1, model_name='author', name='age',
)
self.assertOperationAttributes(
changes, 'testapp', 0, 2, model_name='author', name='name',
)
self.assertOperationAttributes(
changes, 'testapp', 0, 3, name='author', index_together={('age',)},
)
def test_partly_alter_foo_together(self):
initial_author = ModelState('testapp', 'Author', [
('id', models.AutoField(primary_key=True)),
('name', models.CharField(max_length=200)),
('age', models.IntegerField()),
], {
'unique_together': {('name',), ('age',)},
'index_together': {('name',)},
})
author_reversed_constraints = ModelState('testapp', 'Author', [
('id', models.AutoField(primary_key=True)),
('name', models.CharField(max_length=200)),
('age', models.IntegerField()),
], {
'unique_together': {('age',)},
'index_together': {('name',), ('age',)},
})
changes = self.get_changes([initial_author], [author_reversed_constraints])
self.assertNumberMigrations(changes, 'testapp', 1)
self.assertOperationTypes(changes, 'testapp', 0, [
'AlterUniqueTogether',
'AlterIndexTogether',
])
self.assertOperationAttributes(
changes, 'testapp', 0, 0, name='author', unique_together={('age',)},
)
self.assertOperationAttributes(
changes, 'testapp', 0, 1, name='author',
index_together={('name',), ('age',)},
)
def test_rename_field_and_foo_together(self): def test_rename_field_and_foo_together(self):
""" """
@ -1635,11 +1742,27 @@ class AutodetectorTests(TestCase):
) )
# Right number/type of migrations? # Right number/type of migrations?
self.assertNumberMigrations(changes, "otherapp", 1) self.assertNumberMigrations(changes, "otherapp", 1)
self.assertOperationTypes(changes, "otherapp", 0, ["RenameField", "AlterUniqueTogether", "AlterIndexTogether"]) self.assertOperationTypes(changes, 'otherapp', 0, [
self.assertOperationAttributes(changes, "otherapp", 0, 1, name="book", unique_together={ 'RenameField',
("title", "newfield2") 'AlterUniqueTogether',
}) 'AlterIndexTogether',
self.assertOperationAttributes(changes, "otherapp", 0, 2, name="book", index_together={("title", "newfield2")}) 'AlterUniqueTogether',
'AlterIndexTogether',
])
self.assertOperationAttributes(
changes, 'otherapp', 0, 1, name='book', unique_together=set(),
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 2, name='book', index_together=set(),
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 3, name='book',
unique_together={('title', 'newfield2')},
)
self.assertOperationAttributes(
changes, 'otherapp', 0, 4, name='book',
index_together={('title', 'newfield2')},
)
def test_proxy(self): def test_proxy(self):
"""The autodetector correctly deals with proxy models.""" """The autodetector correctly deals with proxy models."""