Merge pull request #516 from tommyzli/master

Addresses #515 - DateType's to_native now converts datetimes into dates
schematics · Sep 19, 2017 · d04cb32 · d04cb32
2 parents ca9b5fc + 27ea4f6
commit d04cb32
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/schematics/types/base.py b/schematics/types/base.py
@@ -156,7 +156,7 @@ class may be added to subclasses of ``Model`` to define a model schema.
     """
 
     primitive_type = None
-    native_type = None 
+    native_type = None
 
     MESSAGES = {
         'required': _("This field is required."),
@@ -677,6 +677,8 @@ def _mock(self, context=None):
         )
 
     def to_native(self, value, context=None):
+        if isinstance(value, datetime.datetime):
+            return value.date()
         if isinstance(value, datetime.date):
             return value
 

diff --git a/tests/test_types.py b/tests/test_types.py
@@ -66,6 +66,8 @@ def test_date():
 
     assert date_type.to_native(today) is today
 
+    assert date_type.to_native(datetime.datetime(2013, 3, 1)) == today
+
     assert type(date_type._mock()) is datetime.date