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['type'] == 'time'else ExerciseType.REPS;
            self.exercises.append(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'))