store timezone aware datetimes instead of strings

refs #11

Author: timgraham

README.md

@@ -54,6 +54,8 @@ DATABASES = {
 
 - Queries with joins aren't supported.
 
+- `DateTimeField` doesn't support microsecond precision.
+
 ## Troubleshooting
 
 TODO

django_mongodb/base.py

@@ -91,7 +91,10 @@ def _connect(self):
             self.operation_flags = {"save": flags, "delete": flags, "update": flags}
 
         self.connection = MongoClient(
-            host=settings_dict["HOST"] or None, port=int(settings_dict["PORT"] or 27017), **options
+            host=settings_dict["HOST"] or None,
+            port=int(settings_dict["PORT"] or 27017),
+            tz_aware=True,
+            **options,
         )
         db_name = settings_dict["NAME"]
         if db_name:

django_mongodb/features.py

@@ -158,4 +158,13 @@ class DatabaseFeatures(BaseDatabaseFeatures):
             "timezones.tests.NewDatabaseTests.test_cursor_explicit_time_zone",
             "timezones.tests.NewDatabaseTests.test_raw_sql",
         },
+        "BSON Date type doesn't support microsecond precision.": {
+            "basic.tests.ModelRefreshTests.test_refresh_unsaved",
+            "basic.tests.ModelTest.test_microsecond_precision",
+            "timezones.tests.LegacyDatabaseTests.test_auto_now_and_auto_now_add",
+            "timezones.tests.LegacyDatabaseTests.test_aware_datetime_in_local_timezone_with_microsecond",
+            "timezones.tests.LegacyDatabaseTests.test_naive_datetime_with_microsecond",
+            "timezones.tests.NewDatabaseTests.test_aware_datetime_in_local_timezone_with_microsecond",
+            "timezones.tests.NewDatabaseTests.test_naive_datetime_with_microsecond",
+        },
     }

django_mongodb/operations.py

@@ -10,13 +10,19 @@
 class DatabaseOperations(BaseDatabaseOperations):
     compiler_module = "django_mongodb.compiler"
 
+    def adapt_datetimefield_value(self, value):
+        if not settings.USE_TZ and timezone.is_naive(value):
+            value = timezone.make_aware(value)
+        return value
+
     def get_db_converters(self, expression):
         converters = super().get_db_converters(expression)
         internal_type = expression.output_field.get_internal_type()
         if internal_type == "DateField":
             converters.append(self.convert_datefield_value)
         elif internal_type == "DateTimeField":
-            converters.append(self.convert_datetimefield_value)
+            if not settings.USE_TZ:
+                converters.append(self.convert_datetimefield_value)
         elif internal_type == "DecimalField":
             converters.append(self.convert_decimalfield_value)
         elif internal_type == "TimeField":
@@ -32,10 +38,8 @@ def convert_datefield_value(self, value, expression, connection):
 
     def convert_datetimefield_value(self, value, expression, connection):
         if value is not None:
-            value = datetime.datetime.fromisoformat(value)
-            if not settings.USE_TZ and timezone.is_aware(value):
-                # Django expects naive datetimes when settings.USE_TZ is False.
-                value = timezone.make_naive(value)
+            # Django expects naive datetimes when settings.USE_TZ is False.
+            value = timezone.make_naive(value)
         return value
 
     def convert_decimalfield_value(self, value, expression, connection):