From 4a1f8efc162b33079cbf84c0e6ead89b1f737796 Mon Sep 17 00:00:00 2001
From: ks129 <45097959+ks129@users.noreply.github.com>
Date: Mon, 30 Nov 2020 18:59:45 +0200
Subject: Add ObjectId type for MongoDB Pydantic models

---
 backend/models/_object_id.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 backend/models/_object_id.py

(limited to 'backend/models')

diff --git a/backend/models/_object_id.py b/backend/models/_object_id.py
new file mode 100644
index 0000000..f0e47cf
--- /dev/null
+++ b/backend/models/_object_id.py
@@ -0,0 +1,23 @@
+import typing as t
+from bson import ObjectId as OriginalObjectId
+
+
+class ObjectId(OriginalObjectId):
+    """ObjectId implementation for Pydantic."""
+
+    @classmethod
+    def __get_validators__(cls) -> t.Generator[t.Callable, None, None]:
+        """Get validators for Pydantic."""
+        yield cls.validate
+
+    @classmethod
+    def validate(cls, value: t.Any) -> t.Optional["ObjectId"]:
+        """Checks value validity to become ObjectId and if valid, return ObjectId."""
+        if OriginalObjectId.is_valid(value):
+            raise ValueError(f"Invalid value '{value}' for ObjectId.")
+        return ObjectId(value)
+
+    @classmethod
+    def __modify_schema__(cls, field_schema: t.Dict[str, t.Any]) -> None:
+        """Update data type to string."""
+        field_schema.update(type="string")
-- 
cgit v1.2.3