JSONEncoder增加timedelta类型

xydoublez · Jul 31, 2018 · 843fb61 · 843fb61
1 parent 41461df
commit 843fb61
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/sql/extend_json_encoder.py b/sql/extend_json_encoder.py
@@ -1,7 +1,7 @@
 # -*- coding: UTF-8 -*-
 import simplejson as json
 
-from datetime import datetime, date
+from datetime import datetime, date, timedelta
 from decimal import Decimal
 from functools import singledispatch
 
@@ -36,6 +36,10 @@ def _(o):
     return o.strftime('%Y-%m-%d')
 
 
+@convert.register(timedelta)
+def _(o):
+    return o.total_seconds()
+
 # @convert.register(Decimal)
 # def _(o):
 #     return float(o)
@@ -59,7 +63,9 @@ def default(self, obj):
     'dm': dm,
     'dt': dt,
     'dat': dat,
+    'tl': timedelta(minutes=30),
     'bigint': 988983860501598208
 }
 
-#print(json.dumps(data, cls=ExtendJSONEncoder, bigint_as_string=True))
+# print(json.dumps(data, cls=ExtendJSONEncoder, bigint_as_string=True))
+# print(json.dumps(data, cls=ExtendJSONEncoder, bigint_as_string=True, default=str))