Added db support

1 files changed, 16 insertions, 0 deletions

diff --git a/utils/json.py b/utils/json.py
new file mode 100644
index 0000000..05d66cb
--- /dev/null
+++ b/utils/json.py
@@ -0,0 +1,16 @@
+import json
+from json import JSONEncoder
+
+from pydantic import BaseModel
+
+
+class Encoder(JSONEncoder):
+    def default(self, o):
+        if isinstance(o, BaseModel):
+            return o.model_dump()
+
+        return super().default(o)
+
+
+def json_serializer(item: any):
+    return json.dumps(item, cls=Encoder)