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))