From dee4d23f7e703aec2d1244e4facbf7f4c88deed5 Mon Sep 17 00:00:00 2001 From: Gavin Wahl Date: Tue, 2 Dec 2014 18:52:58 -0700 Subject: [PATCH] Fixed #23950 -- Prevented calling deconstruct on classes in MigrationWriter. --- django/db/migrations/writer.py | 28 ++++++++++++++-------------- docs/releases/1.7.2.txt | 3 +++ tests/migrations/test_writer.py | 11 +++++++++++ 3 files changed, 28 insertions(+), 14 deletions(-) diff --git a/django/db/migrations/writer.py b/django/db/migrations/writer.py index 1e093778a5d..dec0eae33df 100644 --- a/django/db/migrations/writer.py +++ b/django/db/migrations/writer.py @@ -330,6 +330,20 @@ class MigrationWriter(object): elif isinstance(value, models.Field): attr_name, path, args, kwargs = value.deconstruct() return cls.serialize_deconstructed(path, args, kwargs) + # Classes + elif isinstance(value, type): + special_cases = [ + (models.Model, "models.Model", []), + ] + for case, string, imports in special_cases: + if case is value: + return string, set(imports) + if hasattr(value, "__module__"): + module = value.__module__ + if module == six.moves.builtins.__name__: + return value.__name__, set() + else: + return "%s.%s" % (module, value.__name__), {"import %s" % module} # Anything that knows how to deconstruct itself. elif hasattr(value, 'deconstruct'): return cls.serialize_deconstructed(*value.deconstruct()) @@ -364,20 +378,6 @@ class MigrationWriter(object): "https://docs.djangoproject.com/en/dev/topics/migrations/#serializing-values" % (value.__name__, module_name)) return "%s.%s" % (module_name, value.__name__), {"import %s" % module_name} - # Classes - elif isinstance(value, type): - special_cases = [ - (models.Model, "models.Model", []), - ] - for case, string, imports in special_cases: - if case is value: - return string, set(imports) - if hasattr(value, "__module__"): - module = value.__module__ - if module == six.moves.builtins.__name__: - return value.__name__, set() - else: - return "%s.%s" % (module, value.__name__), {"import %s" % module} # Other iterables elif isinstance(value, collections.Iterable): imports = set() diff --git a/docs/releases/1.7.2.txt b/docs/releases/1.7.2.txt index 39e89a25d18..a537207924a 100644 --- a/docs/releases/1.7.2.txt +++ b/docs/releases/1.7.2.txt @@ -101,3 +101,6 @@ Bugfixes * Fixed ``runserver`` crash when socket error message contained Unicode characters (:ticket:`23946`). + +* Fixed serialization of ``type`` when adding a ``deconstruct()`` method + (:ticket:`23950`). diff --git a/tests/migrations/test_writer.py b/tests/migrations/test_writer.py index 1dcc8e641af..618f8df9258 100644 --- a/tests/migrations/test_writer.py +++ b/tests/migrations/test_writer.py @@ -340,3 +340,14 @@ class WriterTests(TestCase): fixed_offset_datetime = datetime.datetime(2014, 1, 1, 1, 1, tzinfo=FixedOffset(180)) self.assertEqual(MigrationWriter.serialize_datetime(fixed_offset_datetime), "datetime.datetime(2013, 12, 31, 22, 1, tzinfo=utc)") + + def test_deconstruct_class_arguments(self): + # Yes, it doesn't make sense to use a class as a default for a + # CharField. It does make sense for custom fields though, for example + # an enumfield that takes the enum class as an argument. + class DeconstructableInstances(object): + def deconstruct(self): + return ('DeconstructableInstances', [], {}) + + string = MigrationWriter.serialize(models.CharField(default=DeconstructableInstances))[0] + self.assertEqual(string, "models.CharField(default=migrations.test_writer.DeconstructableInstances)")