1 files changed, 21 insertions, 11 deletions
diff --git a/app/tables/workout.py b/app/tables/workout.py
index 564caf1..75faf6b 100644
--- a/app/tables/workout.py
+++ b/app/tables/workout.py
@@ -1,39 +1,42 @@
 from app import db, login_manager
 import enum
 
+
 class ExerciseType(enum.Enum):
     TIME = "Seconds"
     REPS = "Reps"
 
+
 # Represents an individual exercise
 class Exercise(db.Model):
-    __tablename__ = 'exercises'
+    __tablename__ = "exercises"
     id = db.Column(db.Integer, primary_key=True)
     name = db.Column(db.String(100), nullable=False)
     sets = db.Column(db.Integer)
     units = db.Column(db.Integer)
     type = db.Column(db.Enum(ExerciseType))
     # Workout Relationship
-    workout_id = db.Column(db.Integer, db.ForeignKey('workouts.id'))
+    workout_id = db.Column(db.Integer, db.ForeignKey("workouts.id"))
 
     def __init__(self, name, sets, units, type):
         self.name = name
         self.sets = sets
-        self.units = units 
+        self.units = units
         self.type = type
 
     def __repr__(self):
-        return f'<Exercise {self.name} {self.sets}x{self.units}>'
+        return f"<Exercise {self.name} {self.sets}x{self.units}>"
+
 
 # Represents a singular workout
 class Workout(db.Model):
-    __tablename__ = 'workouts'
+    __tablename__ = "workouts"
     id = db.Column(db.Integer, primary_key=True)
     name = db.Column(db.String(100), nullable=False)
     # foreign keys
-    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
+    user_id = db.Column(db.Integer, db.ForeignKey("users.id"))
     # relationships
-    exercises = db.relationship("Exercise", backref='workout', lazy='dynamic')
+    exercises = db.relationship("Exercise", backref="workout", lazy="dynamic")
 
     def __init__(self, user, name, exercises):
         self.name = name
@@ -41,14 +44,21 @@ class Workout(db.Model):
 
         # Create exercises
         for exercise in exercises:
-            type = ExerciseType.TIME if exercise['exercise_type'] == 'time' else ExerciseType.REPS;
-            self.exercises.append(Exercise(exercise['exercise_name'], exercise['sets'], exercise['units'], type))
+            type = (
+                ExerciseType.TIME
+                if exercise["exercise_type"] == "time"
+                else ExerciseType.REPS
+            )
+            self.exercises.append(
+                Exercise(exercise["name"], exercise["sets"], exercise["units"], type)
+            )
 
     def __repr__(self):
-        return f'<Workout {self.name}>'
+        return f"<Workout {self.name}>"
+
 
 # Connects workouts to it's exercises
-#class WorkoutExercise(db.Model):
+# 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'))