about summary refs log tree commit diff
path: root/gn2/utility/json
diff options
context:
space:
mode:
Diffstat (limited to 'gn2/utility/json')
-rw-r--r--gn2/utility/json/__init__.py3
-rw-r--r--gn2/utility/json/encoders.py17
2 files changed, 20 insertions, 0 deletions
diff --git a/gn2/utility/json/__init__.py b/gn2/utility/json/__init__.py
new file mode 100644
index 00000000..b1141a34
--- /dev/null
+++ b/gn2/utility/json/__init__.py
@@ -0,0 +1,3 @@
+"""Local JSON utilities."""
+
+from .encoders import CustomJSONEncoder
diff --git a/gn2/utility/json/encoders.py b/gn2/utility/json/encoders.py
new file mode 100644
index 00000000..7c5839ac
--- /dev/null
+++ b/gn2/utility/json/encoders.py
@@ -0,0 +1,17 @@
+"""Custom JSON encoders"""
+from uuid import UUID
+from json import JSONEncoder
+
+# Do not use this `__ENCODERS__` variable outside of this module.
+__ENCODERS__ = {
+    UUID: lambda obj: str(obj)
+}
+
+class CustomJSONEncoder(JSONEncoder):
+    """Custom JSONEncoder class."""
+    def default(self, obj):
+        """Serialise `obj` to a JSON representation."""
+        obj_type = type(obj)
+        if obj_type in __ENCODERS__:
+            return __ENCODERS__[obj_type](obj)
+        return JSONEncoder.default(self, obj)