about summary refs log tree commit diff
path: root/wqflask/utility/json/encoders.py
diff options
context:
space:
mode:
Diffstat (limited to 'wqflask/utility/json/encoders.py')
-rw-r--r--wqflask/utility/json/encoders.py17
1 files changed, 17 insertions, 0 deletions
diff --git a/wqflask/utility/json/encoders.py b/wqflask/utility/json/encoders.py
new file mode 100644
index 00000000..7c5839ac
--- /dev/null
+++ b/wqflask/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)