p31829507
/
django
mirror of https://github.com/django/django.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Refs #26709 -- Checked allow_migrate_model() in Add/RemoveIndex operations.

This commit is contained in:
Akshesh 2016-07-07 12:02:00 +05:30 committed by Tim Graham
parent 3551fb537f
commit fc3ac65735
2 changed files with 30 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
django/db/migrations/operations/models.py
View File

@ -772,11 +772,13 @@ class AddIndex(IndexOperation):
def database_forwards(self, app_label, schema_editor, from_state, to_state): def database_forwards(self, app_label, schema_editor, from_state, to_state):
model = to_state.apps.get_model(app_label, self.model_name) model = to_state.apps.get_model(app_label, self.model_name)
schema_editor.add_index(model, self.index) if self.allow_migrate_model(schema_editor.connection.alias, model):
schema_editor.add_index(model, self.index)
def database_backwards(self, app_label, schema_editor, from_state, to_state): def database_backwards(self, app_label, schema_editor, from_state, to_state):
model = from_state.apps.get_model(app_label, self.model_name) model = from_state.apps.get_model(app_label, self.model_name)
schema_editor.remove_index(model, self.index) if self.allow_migrate_model(schema_editor.connection.alias, model):
schema_editor.remove_index(model, self.index)
def deconstruct(self): def deconstruct(self):
kwargs = { kwargs = {
@ -812,15 +814,17 @@ class RemoveIndex(IndexOperation):
def database_forwards(self, app_label, schema_editor, from_state, to_state): def database_forwards(self, app_label, schema_editor, from_state, to_state):
model = from_state.apps.get_model(app_label, self.model_name) model = from_state.apps.get_model(app_label, self.model_name)
from_model_state = from_state.models[app_label, self.model_name_lower] if self.allow_migrate_model(schema_editor.connection.alias, model):
index = from_model_state.get_index_by_name(self.name) from_model_state = from_state.models[app_label, self.model_name_lower]
schema_editor.remove_index(model, index) index = from_model_state.get_index_by_name(self.name)
schema_editor.remove_index(model, index)
def database_backwards(self, app_label, schema_editor, from_state, to_state): def database_backwards(self, app_label, schema_editor, from_state, to_state):
model = to_state.apps.get_model(app_label, self.model_name) model = to_state.apps.get_model(app_label, self.model_name)
to_model_state = to_state.models[app_label, self.model_name_lower] if self.allow_migrate_model(schema_editor.connection.alias, model):
index = to_model_state.get_index_by_name(self.name) to_model_state = to_state.models[app_label, self.model_name_lower]
schema_editor.add_index(model, index) index = to_model_state.get_index_by_name(self.name)
schema_editor.add_index(model, index)
def deconstruct(self): def deconstruct(self):
kwargs = { kwargs = {

18
tests/migrations/test_operations.py
View File

@ -2347,3 +2347,21 @@ class SwappableOperationTests(OperationTestBase):
with connection.schema_editor() as editor: with connection.schema_editor() as editor:
operation.database_backwards("test_adfligsw", editor, new_state, project_state) operation.database_backwards("test_adfligsw", editor, new_state, project_state)
self.assertTableNotExists("test_adfligsw_pony") self.assertTableNotExists("test_adfligsw_pony")
@override_settings(TEST_SWAP_MODEL='migrations.SomeFakeModel')
def test_indexes_ignore_swapped(self):
"""
Add/RemoveIndex operations ignore swapped models.
"""
operation = migrations.AddIndex('Pony', models.Index(fields=['pink'], name='my_name_idx'))
project_state, new_state = self.make_test_state('test_adinigsw', operation)
with connection.schema_editor() as editor:
# No database queries should be run for swapped models
operation.database_forwards('test_adinigsw', editor, project_state, new_state)
operation.database_backwards('test_adinigsw', editor, new_state, project_state)
operation = migrations.RemoveIndex('Pony', models.Index(fields=['pink'], name='my_name_idx'))
project_state, new_state = self.make_test_state("test_rminigsw", operation)
with connection.schema_editor() as editor:
operation.database_forwards('test_rminigsw', editor, project_state, new_state)
operation.database_backwards('test_rminigsw', editor, new_state, project_state)