3 files changed, 22 insertions, 8 deletions
diff --git a/app/tables/__init__.py b/app/tables/__init__.py
new file mode 100644
index 0000000..e67eb21
--- /dev/null
+++ b/app/tables/__init__.py
@@ -0,0 +1,2 @@
+from tables.user import User
+from tables.workout import *
diff --git a/app/tables/user.py b/app/tables/user.py
index d384fdb..cfffb80 100644
--- a/app/tables/user.py
+++ b/app/tables/user.py
@@ -13,6 +13,8 @@ class User(db.Model, UserMixin):
     password = db.Column(db.String(80))
     email = db.Column(db.String(120), unique=True, nullable=False)
 
+    workouts = db.relationship("Workout", backref='user', lazy='dynamic')
+
     def __init__(self, username, password, email):
         self.username = username
         self.password = generate_password_hash(password)
diff --git a/app/tables/workout.py b/app/tables/workout.py
index f3a41eb..81f9a68 100644
--- a/app/tables/workout.py
+++ b/app/tables/workout.py
@@ -1,19 +1,29 @@
 from app import db, login_manager
 
 # Represents an individual exercise
-class Exercise(db.Model)
+class Exercise(db.Model):
     __tablename__ = 'exercises'
     id = db.Column(db.Integer, primary_key=True)
 
 # Represents a singular workout
-class Workout(db.Model)
+class Workout(db.Model):
     __tablename__ = 'workouts'
     id = db.Column(db.Integer, primary_key=True)
-    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
+    name = db.Column(db.String(100), nullable=False)
+
+    # User Relationship
+    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
+
+    def __init__(self, user, name, exercises):
+        self.name = name
+        self.user_id = user.id
+
+    def __repr__(self):
+        return f'<Workout {self.name}>'
 
 # Connects workouts to it's exercises
-class WorkoutExercise(db.Model)
-    __table__ = 'workout_exercises'
-    id = db.Column(db.Integer, primary_key=True)
-    workout_id = db.Column(db.Integer, db.ForeignKey('workout.id'))
-    exercise_id = db.Column(db.Integer, db.ForeignKey('exercise.id'))
+#class WorkoutExercise(db.Model):
+#    __table__ = 'workout_exercises'
+#    id = db.Column(db.Integer, primary_key=True)
+#    workout_id = db.Column(db.Integer, db.ForeignKey('workouts.id'))
+#    exercise_id = db.Column(db.Integer, db.ForeignKey('exercises.id'))