From cae7ee8ab4639963d3da7fef6f078a6078340a0b Mon Sep 17 00:00:00 2001
From: Georgios Andreadis <info@gandreadis.com>
Date: Thu, 25 Jun 2020 16:06:22 +0200
Subject: Improve abstraction

---
 opendc/models/model.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 opendc/models/model.py

(limited to 'opendc/models/model.py')

diff --git a/opendc/models/model.py b/opendc/models/model.py
new file mode 100644
index 00000000..d887e845
--- /dev/null
+++ b/opendc/models/model.py
@@ -0,0 +1,31 @@
+from opendc.util.database import DB
+from opendc.util.rest import Response
+
+
+class Model:
+    collection_name = '<specified in subclasses>'
+
+    @classmethod
+    def from_id(cls, _id):
+        return cls(DB.fetch_one({'_id': _id}, Model.collection_name))
+
+    def __init__(self, obj):
+        self.obj = obj
+
+    def validate(self, request_google_id=None):
+        if self.obj is None:
+            return Response(404, f'Not found.')
+
+        return None
+
+    def set_property(self, key, value):
+        self.obj[key] = value
+
+    def insert(self):
+        DB.insert(self.obj, self.collection_name)
+
+    def update(self):
+        DB.update(self.obj['_id'], self.obj, self.collection_name)
+
+    def delete(self):
+        DB.delete_one({'_id': self.obj['_id']}, self.collection_name)
-- 
cgit v1.2.3