Marshalling: Disable option orjson.OPT_PASSTHROUGH_DATETIME

Author: amotl

src/crate/client/http.py

@@ -95,29 +95,17 @@ def cratedb_json_encoder(obj):
     Encoder function for orjson.
 
     https://github.com/ijl/orjson#default
-    https://github.com/ijl/orjson#opt_passthrough_datetime
     """
-    if isinstance(obj, (Decimal, UUID)):
+    if isinstance(obj, (Decimal,)):
         return str(obj)
-    if isinstance(obj, datetime):
-        if obj.tzinfo is not None:
-            delta = obj - epoch_aware
-        else:
-            delta = obj - epoch_naive
-        return int(
-            delta.microseconds / 1000.0
-            + (delta.seconds + delta.days * 24 * 3600) * 1000.0
-        )
-    if isinstance(obj, date):
-        return calendar.timegm(obj.timetuple()) * 1000
     return obj
 
 
 def json_dumps(obj):
     return orjson.dumps(
         obj,
         default=cratedb_json_encoder,
-        option=(orjson.OPT_PASSTHROUGH_DATETIME | orjson.OPT_SERIALIZE_NUMPY),
+        option=orjson.OPT_SERIALIZE_NUMPY,
     )
 
 

tests/client/test_http.py

@@ -318,7 +318,7 @@ def test_datetime_is_converted_to_ts(self, request):
         # convert string to dict
         # because the order of the keys isn't deterministic
         data = json.loads(request.call_args[1]["data"])
-        self.assertEqual(data["args"], [1425108700000])
+        self.assertEqual(data["args"], ["2015-02-28T07:31:40"])
         client.close()
 
     @patch(REQUEST, autospec=True)
@@ -329,7 +329,7 @@ def test_date_is_converted_to_ts(self, request):
         day = dt.date(2016, 4, 21)
         client.sql("insert into users (dt) values (?)", (day,))
         data = json.loads(request.call_args[1]["data"])
-        self.assertEqual(data["args"], [1461196800000])
+        self.assertEqual(data["args"], ["2016-04-21"])
         client.close()
 
     def test_socket_options_contain_keepalive(self):