diff options
| author | Georgios Andreadis <info@gandreadis.com> | 2020-06-25 16:06:22 +0200 |
|---|---|---|
| committer | Georgios Andreadis <info@gandreadis.com> | 2020-06-25 16:06:22 +0200 |
| commit | cae7ee8ab4639963d3da7fef6f078a6078340a0b (patch) | |
| tree | 3777700d19321c3acb043e4ab99bd333af35fc97 /opendc/models/model.py | |
| parent | ff31029e732dd56743690dc608f535163a0548ff (diff) | |
Improve abstraction
Diffstat (limited to 'opendc/models/model.py')
| -rw-r--r-- | opendc/models/model.py | 31 |
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) |
