Initial commit

1 files changed, 27 insertions, 0 deletions

diff --git a/opendc/models/allowed_object.py b/opendc/models/allowed_object.py
new file mode 100644
index 00000000..acd58e94
--- /dev/null
+++ b/opendc/models/allowed_object.py
@@ -0,0 +1,27 @@
+from opendc.models.model import Model
+
+class AllowedObject(Model):
+
+    JSON_TO_PYTHON_DICT = {
+        'AllowedObject': {
+            'roomType': 'room_type',
+            'objectType': 'object_type'
+        }
+    }
+
+    TABLE_NAME = 'allowed_objects'
+    COLUMNS = ['room_type', 'object_type']
+    COLUMNS_PRIMARY_KEY = ['room_type', 'object_type']
+
+    def google_id_has_at_least(self, google_id, authorization_level):
+        """Return True if the user has at least the given auth level over this AllowedObject."""
+
+        if authorization_level in ['EDIT', 'OWN']:
+            return False
+
+        return True
+
+    def to_JSON(self):
+        """Return a JSON representation of this AllowedObject."""
+
+        return self.object_type