add json_serialize in utils

Author: supercoderhawk

pysenal/utils/utils.py

@@ -2,6 +2,7 @@
 import os
 import copy
 from collections import Iterable
+from decimal import Decimal
 
 
 def get_chunk(l, n):
@@ -111,3 +112,24 @@ def index(l, val, default=-1):
         return default
     else:
         return l.index(val)
+
+
+def json_serialize(obj):
+    """
+    add serialize method used in json.dumps or json.dump
+    :param obj: input obj
+    :return: string representation for json.dump or json.dumps
+    """
+    if isinstance(obj, Decimal):
+        numerator, denumerator = obj.as_integer_ratio()
+        if denumerator == 1:
+            return str(numerator)
+        else:
+            return str(obj)
+    elif isinstance(obj, bytes):
+        return str(obj)
+    else:
+        try:
+            return str(obj)
+        except:
+            raise TypeError(repr(obj) + ' is not JSON serializable')

tests/utils/test_utils.py

@@ -2,6 +2,7 @@
 import shutil
 import tempfile
 from types import GeneratorType
+from decimal import Decimal
 import pytest
 from pysenal.utils.utils import *
 
@@ -103,4 +104,11 @@ def test_index():
     assert index(values, 6) == -1
     assert index(values, 6, default=100) == 100
     with pytest.raises(TypeError):
-        index(10,10)
+        index(10, 10)
+
+
+def test_json_serialize():
+    assert json_serialize(Decimal('3.1415926')) == '3.1415926'
+    assert json_serialize(Decimal('11.0')) == '11'
+    assert json_serialize(b'123') == "b'123'"
+    assert json_serialize(11) == '11'