from superset.models import Superset
from superset.models import SupersetExercise
from exercise.models import Exercise

def get_first_up_superset(workout):
        first_up_superset = Superset()
        first_up_superset.rounds = 1
        first_up_superset.order = 0
        first_up_superset.estimated_time = 15
        first_up_superset.name = "Lets Go"
        first_up_superset.workout = workout
        return first_up_superset

def get_first_up_superset_exercise(superset):
    try:
        exercise = Exercise.objects.get(name="First Up")
    except Exercise.DoesNotExist:
        exercise = None

    if exercise is None:
        return None

    first_up_superset_exercise = SupersetExercise()
    first_up_superset_exercise.exercise = exercise
    first_up_superset_exercise.duration = 15
    first_up_superset_exercise.order = 1
    # Build overrides as a dict instead of mutating the ORM object in memory
    first_up_superset_exercise._display_overrides = {
        'side': 'Chill',
        'joints_used': '',
        'movement_patterns': '',
        'equipment_required': '',
        'muscle_groups': '',
    }
    if superset is not None:
        first_up_superset_exercise.superset = superset
    return first_up_superset_exercise