Added form classes for "create workout" form

3 files changed, 28 insertions, 0 deletions

diff --git a/app/forms/__init__.py b/app/forms/__init__.py
new file mode 100644
index 0000000..b904c84
--- /dev/null
+++ b/app/forms/__init__.py
@@ -0,0 +1,2 @@
+from forms.user import *
+from forms.workout import *
diff --git a/app/forms/user.py b/app/forms/user.py
new file mode 100644
index 0000000..12e4bf8
--- /dev/null
+++ b/app/forms/user.py
@@ -0,0 +1,13 @@
+from flask_wtf import FlaskForm
+from wtforms import StringField, PasswordField, BooleanField
+from wtforms.validators import DataRequired, Email
+
+class LoginForm(FlaskForm):
+    username = StringField("username", validators=[DataRequired()])
+    password = PasswordField("password", validators=[DataRequired()])
+    #remember_me = BooleanField()
+
+class RegisterForm(FlaskForm):
+    username = StringField("username", validators=[DataRequired()])
+    password = PasswordField("password", validators=[DataRequired()])
+    email = StringField("email", validators=[DataRequired(), Email()])
diff --git a/app/forms/workout.py b/app/forms/workout.py
new file mode 100644
index 0000000..4a96b32
--- /dev/null
+++ b/app/forms/workout.py
@@ -0,0 +1,13 @@
+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):
+    name = StringField("name",  [DataRequired()])
+    sets = IntegerField("sets", [DataRequired()])
+    reps = IntegerField("reps", [DataRequired()])
+
+class WorkoutCreateForm(FlaskForm):
+    name = StringField("name", [DataRequired()])
+    exercises = FieldList(FormField(ExerciseForm), min_entries=1)