From ff31029e732dd56743690dc608f535163a0548ff Mon Sep 17 00:00:00 2001
From: Georgios Andreadis <info@gandreadis.com>
Date: Thu, 25 Jun 2020 14:48:24 +0200
Subject: Experiment

---
 opendc/models_old/user.py | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 opendc/models_old/user.py

(limited to 'opendc/models_old/user.py')

diff --git a/opendc/models_old/user.py b/opendc/models_old/user.py
new file mode 100644
index 00000000..657d5019
--- /dev/null
+++ b/opendc/models_old/user.py
@@ -0,0 +1,47 @@
+from opendc.models_old.model import Model
+
+
+class User(Model):
+    JSON_TO_PYTHON_DICT = {
+        'User': {
+            'id': 'id',
+            'googleId': 'google_id',
+            'email': 'email',
+            'givenName': 'given_name',
+            'familyName': 'family_name'
+        }
+    }
+
+    COLLECTION_NAME = 'users'
+    COLUMNS = ['id', 'google_id', 'email', 'given_name', 'family_name']
+    COLUMNS_PRIMARY_KEY = ['id']
+
+    @classmethod
+    def from_google_id(cls, google_id):
+        """Initialize a User by fetching them by their google id."""
+
+        user = cls._from_database('SELECT * FROM users WHERE google_id = %s', (google_id, ))
+
+        if user is not None:
+            return user
+
+        return User()
+
+    @classmethod
+    def from_email(cls, email):
+        """Initialize a User by fetching them by their email."""
+
+        user = cls._from_database('SELECT * FROM users WHERE email = %s', (email, ))
+
+        if user is not None:
+            return user
+
+        return User()
+
+    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 User."""
+
+        if authorization_level in ['EDIT', 'OWN']:
+            return google_id == self.google_id
+
+        return True
-- 
cgit v1.2.3