Browse Source

Working on new models

JoostSijm 6 years ago
parent
commit
4113dfcdd8
1 changed files with 10 additions and 3 deletions
  1. 10 3
      app/models.py

+ 10 - 3
app/models.py

@@ -34,7 +34,6 @@ page_file = db.Table(
 
 class User(db.Model, UserMixin):
     """Model for User"""
-
     id = db.Column(db.Integer, primary_key=True)
     name = db.Column(db.String, unique=True, nullable=False)
     email = db.Column(db.String(255), unique=True)
@@ -64,9 +63,18 @@ class User(db.Model, UserMixin):
         return argon2.check_password_hash(self.password, password)
 
 
+    function_id = db.Column(
+        db.Integer,
+        db.ForeignKey("function.id")
+    )
+    user = db.relationship(
+        "Function",
+        backref=db.backref("users", lazy="dynamic")
+    )
+
+
 class Page(db.Model):
     """Model for Page"""
-
     id = db.Column(db.Integer, primary_key=True)
     title = db.Column(db.String, nullable=False)
     datetime = db.Column(db.DateTime, default=datetime.utcnow)
@@ -113,7 +121,6 @@ class Page(db.Model):
 
 class File(db.Model):
     """Model for Page"""
-
     id = db.Column(db.Integer, primary_key=True)
     title = db.Column(db.String, nullable=False)
     datetime = db.Column(db.DateTime, default=datetime.utcnow)