Workout table and user relationship works

1 files changed, 18 insertions, 8 deletions

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'))