Fixed #22331 -- Made MigrationAutodetector ignore unmanaged models.
This commit reverts 69d4b1c
and tackle the issue from a different angle.
Models remain present in the project state, but are now ignored by the
autodetector.
This commit is contained in:
parent
378359de1c
commit
42336c84a0
|
@ -51,16 +51,17 @@ class MigrationAutodetector(object):
|
|||
new_apps = self.to_state.render()
|
||||
# Prepare lists of old/new model keys that we care about
|
||||
# (i.e. ignoring proxy ones)
|
||||
old_model_keys = [
|
||||
(al, mn)
|
||||
for al, mn in self.from_state.models.keys()
|
||||
if not old_apps.get_model(al, mn)._meta.proxy
|
||||
]
|
||||
new_model_keys = [
|
||||
(al, mn)
|
||||
for al, mn in self.to_state.models.keys()
|
||||
if not new_apps.get_model(al, mn)._meta.proxy
|
||||
]
|
||||
old_model_keys = []
|
||||
for al, mn in self.from_state.models.keys():
|
||||
model = old_apps.get_model(al, mn)
|
||||
if not model._meta.proxy and model._meta.managed:
|
||||
old_model_keys.append((al, mn))
|
||||
|
||||
new_model_keys = []
|
||||
for al, mn in self.to_state.models.keys():
|
||||
model = new_apps.get_model(al, mn)
|
||||
if not model._meta.proxy and model._meta.managed:
|
||||
new_model_keys.append((al, mn))
|
||||
|
||||
def _rel_agnostic_fields_def(fields):
|
||||
"""
|
||||
|
|
|
@ -58,9 +58,8 @@ class ProjectState(object):
|
|||
"Takes in an Apps and returns a ProjectState matching it"
|
||||
app_models = {}
|
||||
for model in apps.get_models():
|
||||
if model._meta.managed:
|
||||
model_state = ModelState.from_model(model)
|
||||
app_models[(model_state.app_label, model_state.name.lower())] = model_state
|
||||
model_state = ModelState.from_model(model)
|
||||
app_models[(model_state.app_label, model_state.name.lower())] = model_state
|
||||
return cls(app_models)
|
||||
|
||||
def __eq__(self, other):
|
||||
|
|
|
@ -24,6 +24,8 @@ class AutodetectorTests(TestCase):
|
|||
author_with_custom_user = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("user", models.ForeignKey("thirdapp.CustomUser"))])
|
||||
author_proxy = ModelState("testapp", "AuthorProxy", [], {"proxy": True}, ("testapp.author", ))
|
||||
author_proxy_notproxy = ModelState("testapp", "AuthorProxy", [], {}, ("testapp.author", ))
|
||||
author_unmanaged = ModelState("testapp", "AuthorUnmanaged", [], {"managed": False}, ("testapp.author", ))
|
||||
author_unmanaged_managed = ModelState("testapp", "AuthorUnmanaged", [], {}, ("testapp.author", ))
|
||||
publisher = ModelState("testapp", "Publisher", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=100))])
|
||||
publisher_with_author = ModelState("testapp", "Publisher", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("testapp.Author")), ("name", models.CharField(max_length=100))])
|
||||
publisher_with_book = ModelState("testapp", "Publisher", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("otherapp.Book")), ("name", models.CharField(max_length=100))])
|
||||
|
@ -488,6 +490,31 @@ class AutodetectorTests(TestCase):
|
|||
self.assertEqual(action.__class__.__name__, "CreateModel")
|
||||
self.assertEqual(action.name, "AuthorProxy")
|
||||
|
||||
def test_unmanaged_ignorance(self):
|
||||
"Tests that the autodetector correctly ignores managed models"
|
||||
# First, we test adding an unmanaged model
|
||||
before = self.make_project_state([self.author_empty])
|
||||
after = self.make_project_state([self.author_empty, self.author_unmanaged])
|
||||
autodetector = MigrationAutodetector(before, after)
|
||||
changes = autodetector._detect_changes()
|
||||
# Right number of migrations?
|
||||
self.assertEqual(len(changes), 0)
|
||||
|
||||
# Now, we test turning an unmanaged model into a managed model
|
||||
before = self.make_project_state([self.author_empty, self.author_unmanaged])
|
||||
after = self.make_project_state([self.author_empty, self.author_unmanaged_managed])
|
||||
autodetector = MigrationAutodetector(before, after)
|
||||
changes = autodetector._detect_changes()
|
||||
# Right number of migrations?
|
||||
self.assertEqual(len(changes['testapp']), 1)
|
||||
# Right number of actions?
|
||||
migration = changes['testapp'][0]
|
||||
self.assertEqual(len(migration.operations), 1)
|
||||
# Right action?
|
||||
action = migration.operations[0]
|
||||
self.assertEqual(action.__class__.__name__, "CreateModel")
|
||||
self.assertEqual(action.name, "AuthorUnmanaged")
|
||||
|
||||
@override_settings(AUTH_USER_MODEL="thirdapp.CustomUser")
|
||||
def test_swappable(self):
|
||||
before = self.make_project_state([self.custom_user])
|
||||
|
|
|
@ -52,22 +52,11 @@ class StateTests(TestCase):
|
|||
verbose_name = "tome"
|
||||
db_table = "test_tome"
|
||||
|
||||
class Unmanaged(models.Model):
|
||||
title = models.CharField(max_length=1000)
|
||||
|
||||
class Meta:
|
||||
app_label = "migrations"
|
||||
apps = new_apps
|
||||
managed = False
|
||||
|
||||
project_state = ProjectState.from_apps(new_apps)
|
||||
author_state = project_state.models['migrations', 'author']
|
||||
author_proxy_state = project_state.models['migrations', 'authorproxy']
|
||||
sub_author_state = project_state.models['migrations', 'subauthor']
|
||||
book_state = project_state.models['migrations', 'book']
|
||||
# unmanaged models should not appear in migrations
|
||||
with self.assertRaises(KeyError):
|
||||
project_state.models['migrations', 'unmanaged']
|
||||
|
||||
self.assertEqual(author_state.app_label, "migrations")
|
||||
self.assertEqual(author_state.name, "Author")
|
||||
|
|
Loading…
Reference in New Issue