diff options
Diffstat (limited to 'app/forms')
| -rw-r--r-- | app/forms/workout.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/app/forms/workout.py b/app/forms/workout.py index 4a96b32..ecc72c8 100644 --- a/app/forms/workout.py +++ b/app/forms/workout.py @@ -2,12 +2,23 @@ from flask_wtf import FlaskForm, Form from wtforms import FieldList, FormField, IntegerField, StringField from wtforms.validators import DataRequired, Email -# Subclass, not used directly -class ExerciseForm(Form): +# Subclasses +class ExerciseCreateForm(Form): name = StringField("name", [DataRequired()]) sets = IntegerField("sets", [DataRequired()]) + units = IntegerField("units", [DataRequired()]) + +class SetForm(Form): + lbs = IntegerField("lbs", [DataRequired()]) reps = IntegerField("reps", [DataRequired()]) +class ExerciseRecordForm(Form): + sets = FieldList(FormField(SetForm)) + +# Actual forms class WorkoutCreateForm(FlaskForm): name = StringField("name", [DataRequired()]) - exercises = FieldList(FormField(ExerciseForm), min_entries=1) + exercises = FieldList(FormField(ExerciseCreateForm), min_entries=1) + +class WorkoutRecordForm(FlaskForm): + exercises = FieldList(FormField(ExerciseRecordForm)) |