1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
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'
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'))
def __init__(self, name, sets, units, type):
self.name = name
self.sets = sets
self.units = units
self.type = type
def __repr__(self):
return f'<Exercise {self.name} {self.sets}x{self.units}>'
# Represents a singular workout
class Workout(db.Model):
__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'))
# relationships
exercises = db.relationship("Exercise", backref='workout', lazy='dynamic')
def __init__(self, user, name, exercises):
self.name = name
self.user_id = user.id
# 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))
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('workouts.id'))
# exercise_id = db.Column(db.Integer, db.ForeignKey('exercises.id'))
|
