Merge pull request #65 from laugengebaeck/fix/enum-encoding

Fix broken enum encoding in JSON export

Author: arneboockmeyer

yaramo/base_element.py

@@ -3,6 +3,8 @@
 
 import simplejson as json
 
+from yaramo.utils.enum_encoder import EnumEncoder
+
 
 class BaseElement(object):
     def __init__(self, uuid: str = None, name: str = None, **kwargs) -> None:
@@ -27,4 +29,4 @@ def to_serializable(self) -> Tuple[dict, dict]:
         return self.__dict__, {}
 
     def to_json(self) -> str:
-        return json.dumps(self.to_serializable()[0], iterable_as_array=True)
+        return json.dumps(self.to_serializable()[0], iterable_as_array=True, cls=EnumEncoder)

yaramo/utils/enum_encoder.py

@@ -0,0 +1,9 @@
+from enum import Enum
+
+import simplejson as json
+
+class EnumEncoder(json.JSONEncoder):
+    def default(self, o):
+        if isinstance(o, Enum):
+            return str(o)
+        return super().default(o)