switch to simplejson

simulate-digital-rail · Feb 9, 2024 · ed2ad54 · ed2ad54
1 parent d82d4fc
commit ed2ad54
Show file tree

Hide file tree

Showing 5 changed files with 112 additions and 14 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,6 +9,7 @@ packages = [{include = "yaramo"}]
 [tool.poetry.dependencies]
 python = "^3.9"
 pyproj = "^3.6.1"
+simplejson = "^3.19.2"
 
 
 [tool.poetry.group.dev.dependencies]

diff --git a/yaramo/base_element.py b/yaramo/base_element.py
@@ -1,9 +1,7 @@
-import json
+import simplejson as json
 from typing import Tuple
 from uuid import uuid4
 
-from yaramo.utils import CustomJSONEncoder
-
 
 class BaseElement(object):
     def __init__(self, uuid: str = None, name: str = None, **kwargs) -> None:
@@ -28,4 +26,4 @@ def to_serializable(self) -> Tuple[dict, dict]:
         return self.__dict__, {}
 
     def to_json(self) -> str:
-        return json.dumps(self.to_serializable()[0], cls=CustomJSONEncoder)
+        return json.dumps(self.to_serializable()[0], iterable_as_array=True)
diff --git a/yaramo/topology.py b/yaramo/topology.py
@@ -1,4 +1,4 @@
-import json
+import simplejson as json
 
 from yaramo.base_element import BaseElement
 from yaramo.edge import Edge

diff --git a/yaramo/utils.py b/yaramo/utils.py