diff --git a/django/db/migrations/writer.py b/django/db/migrations/writer.py index 1b9cdb886e..d9f9c734a2 100644 --- a/django/db/migrations/writer.py +++ b/django/db/migrations/writer.py @@ -11,9 +11,9 @@ import types from django.apps import apps from django.db import models from django.db.migrations.loader import MigrationLoader +from django.utils import datetime_safe, six from django.utils.encoding import force_text from django.utils.functional import Promise -from django.utils import six class SettingsReference(str): @@ -215,10 +215,16 @@ class MigrationWriter(object): elif isinstance(value, datetime.datetime): if value.tzinfo is not None: raise ValueError("Cannot serialize datetime values with timezones. Either use a callable value for default or remove the timezone.") - return repr(value), set(["import datetime"]) + value_repr = repr(value) + if isinstance(value, datetime_safe.datetime): + value_repr = "datetime.%s" % value_repr + return value_repr, set(["import datetime"]) # Dates elif isinstance(value, datetime.date): - return repr(value), set(["import datetime"]) + value_repr = repr(value) + if isinstance(value, datetime_safe.date): + value_repr = "datetime.%s" % value_repr + return value_repr, set(["import datetime"]) # Settings references elif isinstance(value, SettingsReference): return "settings.%s" % value.setting_name, set(["from django.conf import settings"]) diff --git a/tests/migrations/test_writer.py b/tests/migrations/test_writer.py index 8bdce5d8c0..c3889df0e7 100644 --- a/tests/migrations/test_writer.py +++ b/tests/migrations/test_writer.py @@ -10,7 +10,7 @@ from django.db import models, migrations from django.db.migrations.writer import MigrationWriter, SettingsReference from django.test import TestCase from django.conf import settings -from django.utils import six +from django.utils import datetime_safe, six from django.utils.deconstruct import deconstructible from django.utils.translation import ugettext_lazy as _ from django.utils.timezone import get_default_timezone @@ -81,6 +81,14 @@ class WriterTests(TestCase): self.assertSerializedEqual(datetime.date.today) with self.assertRaises(ValueError): self.assertSerializedEqual(datetime.datetime(2012, 1, 1, 1, 1, tzinfo=get_default_timezone())) + safe_date = datetime_safe.date(2014, 3, 31) + string, imports = MigrationWriter.serialize(safe_date) + self.assertEqual(string, repr(datetime.date(2014, 3, 31))) + self.assertEqual(imports, {'import datetime'}) + safe_datetime = datetime_safe.datetime(2014, 3, 31, 16, 4, 31) + string, imports = MigrationWriter.serialize(safe_datetime) + self.assertEqual(string, repr(datetime.datetime(2014, 3, 31, 16, 4, 31))) + self.assertEqual(imports, {'import datetime'}) # Classes validator = RegexValidator(message="hello") string, imports = MigrationWriter.serialize(validator)