add more models

milesgranger · May 10, 2018 · ac43e07 · ac43e07
1 parent 98778e6
commit ac43e07
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/hemlock_highway/server/user_mgmt/models/user.py b/hemlock_highway/server/user_mgmt/models/user.py
@@ -12,14 +12,33 @@ class User(UserMixin, db.Model):
 
     __tablename__ = 'user'
 
-    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    id          = db.Column(db.Integer, primary_key=True, autoincrement=True)
     email       = db.Column(db.String(256), unique=True)
     name_first  = db.Column(db.String(256), unique=False, nullable=True)
     name_last   = db.Column(db.String(256), unique=False, nullable=True)
+    transaction = db.relationship('Transaction')
+    model       = db.relationship('MachineLearningModel')
 
 
+class MachineLearningModel(db.Model):
 
+    __tablename__ = 'ml_model'
 
+    id          = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    user_id     = db.Column(db.Integer, db.ForeignKey(User.id))
+    name        = db.Column(db.String(256), unique=False)
+    trained     = db.Column(db.Boolean, default=False)
+    size_bytes  = db.Column(db.Integer)
+
+
+class Transaction(db.Model):
+
+    __tablename__ = 'transaction'
+
+    id          = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    user_id     = db.Column(db.Integer, db.ForeignKey(User.id))
+    amount      = db.Column(db.Float, unique=False, nullable=False)
+    description = db.Column(db.String(1000), unique=False, nullable=False)
 
 
 class OAuth(OAuthConsumerMixin, db.Model):