Support Decimal in FriendlyEncoder (#12)

Co-authored-by: Roberto Prevato <[email protected]>

Author: tyzhnenko

essentials/json.py

@@ -10,6 +10,7 @@
 from enum import Enum
 from typing import Any
 from uuid import UUID
+from decimal import Decimal
 
 __all__ = ["FriendlyEncoder", "dumps"]
 
@@ -33,6 +34,8 @@ def default(self, obj: Any) -> Any:
                 return base64.urlsafe_b64encode(obj).decode("utf8")
             if isinstance(obj, UUID):
                 return str(obj)
+            if isinstance(obj, Decimal):
+                return str(obj)
             if isinstance(obj, Enum):
                 return obj.value
             if dataclasses.is_dataclass(obj):
@@ -51,7 +54,7 @@ def dumps(
     separators=None,
     default=None,
     sort_keys=False,
-    **kw
+    **kw,
 ) -> str:
     if cls is None:
         cls = FriendlyEncoder
@@ -66,5 +69,5 @@ def dumps(
         separators=separators,
         default=default,
         sort_keys=sort_keys,
-        **kw
+        **kw,
     )

tests/test_json.py

@@ -1,5 +1,6 @@
 from dataclasses import dataclass
 from datetime import date, datetime, time
+from decimal import Decimal
 from enum import Enum, Flag, IntEnum, IntFlag, auto
 from uuid import UUID, uuid4
 
@@ -53,6 +54,7 @@ class Permission(IntFlag):
         ({"value": time(10, 30, 15)}, '{"value": "10:30:15"}'),
         ({"value": date(2016, 3, 26)}, '{"value": "2016-03-26"}'),
         ({"value": datetime(2016, 3, 26, 3, 0, 0)}, '{"value": "2016-03-26T03:00:00"}'),
+        ({"value": Decimal("10.5")}, '{"value": "10.5"}'),
         (
             {"value": UUID("e56fddfc-f85b-4178-869f-a218278a639e")},
             '{"value": "e56fddfc-f85b-4178-869f-a218278a639e"}',