diff options
| author | Marc Tamlyn <marc.tamlyn@gmail.com> | 2014-05-22 12:33:44 +0100 |
|---|---|---|
| committer | Marc Tamlyn <marc.tamlyn@gmail.com> | 2014-05-22 13:23:51 +0100 |
| commit | 6ab67919ac25143444a57f14f6aed76ffe8a6960 (patch) | |
| tree | 1a2dcbef8aadf631a8dc4cf54fcc009be10778e6 | |
| parent | 604162604bf816fa46b0d972621f06de64df6a66 (diff) | |
Made nested deconstruction support both forms of deconstruct()
Nested deconstruction should (silently) handle Field.deconstruct() as
well as other arbitrary deconstructable objects. This allows having a
field in the deconstruction of another field.
| -rw-r--r-- | django/db/migrations/autodetector.py | 9 | ||||
| -rw-r--r-- | tests/migrations/test_autodetector.py | 12 |
2 files changed, 17 insertions, 4 deletions
diff --git a/django/db/migrations/autodetector.py b/django/db/migrations/autodetector.py index cd6f724c75..100783e09e 100644 --- a/django/db/migrations/autodetector.py +++ b/django/db/migrations/autodetector.py @@ -67,20 +67,21 @@ class MigrationAutodetector(object): if not model._meta.proxy and model._meta.managed and al not in self.to_state.real_apps: new_model_keys.append((al, mn)) - def _deep_deconstruct(obj, field=True): + def _deep_deconstruct(obj): """ Recursive deconstruction for a field and its arguments. """ if not hasattr(obj, 'deconstruct'): return obj deconstructed = obj.deconstruct() - if field: + if isinstance(obj, models.Field): + # we have a field which also returns a name deconstructed = deconstructed[1:] name, args, kwargs = deconstructed return ( name, - [_deep_deconstruct(value, field=False) for value in args], - dict([(key, _deep_deconstruct(value, field=False)) + [_deep_deconstruct(value) for value in args], + dict([(key, _deep_deconstruct(value)) for key, value in kwargs.items()]) ) diff --git a/tests/migrations/test_autodetector.py b/tests/migrations/test_autodetector.py index 54361f00ed..88b67d2c23 100644 --- a/tests/migrations/test_autodetector.py +++ b/tests/migrations/test_autodetector.py @@ -28,6 +28,8 @@ class AutodetectorTests(TestCase): author_name_default = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default='Ada Lovelace'))]) author_name_deconstructable_1 = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default=DeconstructableObject()))]) author_name_deconstructable_2 = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default=DeconstructableObject()))]) + author_name_deconstructable_3 = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default=models.IntegerField()))]) + author_name_deconstructable_4 = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default=models.IntegerField()))]) author_with_book = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("book", models.ForeignKey("otherapp.Book"))]) author_renamed_with_book = ModelState("testapp", "Writer", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("book", models.ForeignKey("otherapp.Book"))]) author_with_publisher_string = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("publisher_name", models.CharField(max_length=200))]) @@ -576,6 +578,16 @@ class AutodetectorTests(TestCase): changes = autodetector._detect_changes() self.assertEqual(changes, {}) + def test_deconstruct_field_kwarg(self): + """ + Field instances are handled correctly by nested deconstruction. + """ + before = self.make_project_state([self.author_name_deconstructable_3]) + after = self.make_project_state([self.author_name_deconstructable_4]) + autodetector = MigrationAutodetector(before, after) + changes = autodetector._detect_changes() + self.assertEqual(changes, {}) + def test_replace_string_with_foreignkey(self): """ Adding an FK in the same "spot" as a deleted CharField should work. (#22300). |
