summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMarc Tamlyn <marc.tamlyn@gmail.com>2014-05-22 12:33:44 +0100
committerMarc Tamlyn <marc.tamlyn@gmail.com>2014-05-22 13:23:51 +0100
commit6ab67919ac25143444a57f14f6aed76ffe8a6960 (patch)
tree1a2dcbef8aadf631a8dc4cf54fcc009be10778e6
parent604162604bf816fa46b0d972621f06de64df6a66 (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.py9
-rw-r--r--tests/migrations/test_autodetector.py12
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).