mirror of https://github.com/django/django.git
Fixed #31503 -- Made autodetector remove unique/index_together before altering fields.
This commit is contained in:
parent
c1e4111c74
commit
0314593fe8
|
@ -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,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
@ -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."""
|
||||||
|
|
Loading…
Reference in New Issue