Added db models

edison/models/__init__.py

@@ -0,0 +1,2 @@
+from .token import Token
+from .user import User

edison/models/token.py

@@ -0,0 +1,12 @@
+from edison import db
+
+
+class Token(db.Model):
+    __tablename__ = 'token_blacklist'
+
+    id = db.Column(db.Integer, primary_key=True)
+    jti = db.Column(db.String(150), nullable=False, unique=True)
+    creation_timestamp = db.Column(db.TIMESTAMP(timezone=False), nullable=False)
+
+    def __repr__(self):
+        return f"<Token: jti: {self.jti}, creation time: {self.creation_timestamp}>"

edison/models/user.py

@@ -0,0 +1,25 @@
+from edison import db
+
+
+class User(db.Model):
+    __tablename__ = 'users'
+    __table_args__ = {'extend_existing': True}
+
+    id = db.Column(db.Integer, primary_key=True)
+    username = db.Column(db.String(50), nullable=False, unique=True)
+    password = db.Column(db.String(150), nullable=False)
+    first_name = db.Column(db.String(50), nullable=False)
+    last_name = db.Column(db.String(50), nullable=False)
+    email = db.Column(db.String(150), nullable=False)
+
+    def to_json(self):
+        return {
+            "username": self.username,
+            "first_name": self.first_name,
+            "last_name": self.last_name,
+            "email": self.email
+        }
+
+    def __repr__(self):
+        return f"<User: id = {self.id}, first_name = {self.first_name}, " \
+               f"last_name = {self.last_name}, username = {self.username}, email = {self.email}>"