bugfix in JSONEncoder

list-family · Jul 14, 2020 · 4b9716e · 4b9716e
1 parent 64e36d2
commit 4b9716e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/ansq/__version__.py b/ansq/__version__.py
@@ -1 +1 @@
-__version__ = '0.0.12'
+__version__ = '0.0.13'
diff --git a/ansq/utils.py b/ansq/utils.py
@@ -14,11 +14,10 @@
 
 class JSONEncoder(json.JSONEncoder):
     def default(self, obj) -> str:
-        if isinstance(obj, Enum):
-            return obj.name
-        if isinstance(obj, datetime):
-            return obj.isoformat()
-        return json.JSONEncoder.default(self, obj)
+        try:
+            return convert_to_str(obj)
+        except TypeError:
+            return json.JSONEncoder.default(self, obj)
 
 
 def get_host_port(uri: str) -> Tuple[str, Optional[int]]: