diff options
Diffstat (limited to 'app/routes')
| -rw-r--r-- | app/routes/workout.py | 42 |
1 files changed, 35 insertions, 7 deletions
diff --git a/app/routes/workout.py b/app/routes/workout.py index ffdacb9..0c6116c 100644 --- a/app/routes/workout.py +++ b/app/routes/workout.py @@ -1,8 +1,9 @@ from app import app, db +from datetime import datetime from flask import render_template, redirect, request, url_for, flash from flask_login import current_user, login_required from forms import WorkoutCreateForm, WorkoutRecordForm -from tables import Exercise, User, Workout +from tables import Exercise, User, Workout, WorkoutRecord, SetRecord @app.route("/workout/create", methods=["GET", "POST"]) @@ -25,6 +26,9 @@ def workout_create(): return redirect(url_for("home")) else: flash("Workout with this name already exists", "danger") + else: + print(form) + print(form.errors) return render_template("workout/create.html", form=form) @@ -53,7 +57,7 @@ def workout_edit(): workout_id=workout.id, id=int(entry.data["id"]) ).first() - if not exercise: # NOTE error? + if not exercise: continue # Update exercise @@ -62,8 +66,6 @@ def workout_edit(): exercise.units = entry.data["units"] exercise.type = entry.data["type"] - print(exercise) - # Write changes to database db.session.commit() return redirect(url_for("home")) @@ -96,10 +98,36 @@ def workout_record(): return redirect(url_for("home")) if form.validate_on_submit(): - # Form has been submitted and is valid FIXME - print(form) + # Form has been submitted and is valid + workout_record = WorkoutRecord(workout, datetime.now()) + + # Interate over form exercise entries + for ee in form.exercises.entries: + id = ee.data["id"] + + # And over that exercise's sets + for se in ee.sets.entries: + lbs = se.data["lbs"] + units = se.data["units"] + + if lbs and units: + # Add the set to the workout record + workout_record.sets.append( + SetRecord(lbs=lbs, reps=units, exercise_id=id) + ) + + db.session.add(workout_record) + db.session.commit() + + return redirect(url_for("home")) + else: + # Populate form with data + for exercise in workout.exercises: + form.exercises.append_entry( + {"id": exercise.id, "sets": [{}] * exercise.sets} + ) - return render_template("workout/record.html", workout=workout, form=form) + return render_template("workout/record.html", workout=workout, form=form) @app.route("/workout/select", methods=["GET"]) |