summaryrefslogtreecommitdiff
path: root/opendc/models/model.py
diff options
context:
space:
mode:
authorGeorgios Andreadis <info@gandreadis.com>2020-06-25 16:06:22 +0200
committerGeorgios Andreadis <info@gandreadis.com>2020-06-25 16:06:22 +0200
commitcae7ee8ab4639963d3da7fef6f078a6078340a0b (patch)
tree3777700d19321c3acb043e4ab99bd333af35fc97 /opendc/models/model.py
parentff31029e732dd56743690dc608f535163a0548ff (diff)
Improve abstraction
Diffstat (limited to 'opendc/models/model.py')
-rw-r--r--opendc/models/model.py31
1 files changed, 31 insertions, 0 deletions
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)