summaryrefslogtreecommitdiff
path: root/web-server/opendc/models/model.py
diff options
context:
space:
mode:
authorGeorgios Andreadis <info@gandreadis.com>2020-06-29 16:05:23 +0200
committerFabian Mastenbroek <mail.fabianm@gmail.com>2020-08-24 16:18:36 +0200
commit4f9a40abdc7836345113c047f27fcc96800cb3f5 (patch)
treee443d14e34a884b1a4d9c549f81d51202eddd5f7 /web-server/opendc/models/model.py
parentcd5f7bf3a72913e1602cb4c575e61ac7d5519be0 (diff)
Prepare web-server repository for monorepo
This change prepares the web-server Git repository for the monorepo residing at https://github.com/atlarge-research.com/opendc. To accomodate for this, we move all files into a web-server subdirectory.
Diffstat (limited to 'web-server/opendc/models/model.py')
-rw-r--r--web-server/opendc/models/model.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/web-server/opendc/models/model.py b/web-server/opendc/models/model.py
new file mode 100644
index 00000000..2505ae61
--- /dev/null
+++ b/web-server/opendc/models/model.py
@@ -0,0 +1,30 @@
+from opendc.util.database import DB
+from opendc.util.exceptions import ClientError
+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 check_exists(self):
+ if self.obj is None:
+ raise ClientError(Response(404, 'Not found.'))
+
+ def set_property(self, key, value):
+ self.obj[key] = value
+
+ def insert(self):
+ self.obj = DB.insert(self.obj, self.collection_name)
+
+ def update(self):
+ self.obj = DB.update(self.obj['_id'], self.obj, self.collection_name)
+
+ def delete(self):
+ self.obj = DB.delete_one({'_id': self.obj['_id']}, self.collection_name)