diff options
| author | Georgios Andreadis <info@gandreadis.com> | 2020-06-29 16:05:23 +0200 |
|---|---|---|
| committer | Fabian Mastenbroek <mail.fabianm@gmail.com> | 2020-08-24 16:18:36 +0200 |
| commit | 4f9a40abdc7836345113c047f27fcc96800cb3f5 (patch) | |
| tree | e443d14e34a884b1a4d9c549f81d51202eddd5f7 /web-server/opendc/models/model.py | |
| parent | cd5f7bf3a72913e1602cb4c575e61ac7d5519be0 (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.py | 30 |
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) |
