summaryrefslogtreecommitdiff
path: root/app/forms/workout.py
blob: 45d7f0164dcb3e439cb02537bacef9df88545d03 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
from flask_wtf import FlaskForm, Form
from wtforms.csrf.session import SessionCSRF
from datetime import timedelta
from os import urandom
from wtforms import (
    FieldList,
    FormField,
    HiddenField,
    IntegerField,
    FloatField,
    SelectField,
    StringField,
)
from wtforms.validators import DataRequired, Email, Optional

# /workout/create


class ExerciseEditForm(Form):
    id = HiddenField("id", [Optional()])
    name = StringField("name", [Optional()])
    sets = IntegerField("sets", [Optional()])
    units = FloatField("units", [Optional()])
    type = SelectField(
        "type", [Optional()], choices=[("reps", "Reps"), ("time", "Time")]
    )


class ExerciseCreateForm(Form):
    id = HiddenField("id", [Optional()])
    name = StringField("name", [DataRequired()])
    sets = IntegerField("sets", [DataRequired()])
    units = FloatField("units", [DataRequired()])
    type = SelectField(
        "type", [DataRequired()], choices=[("reps", "Reps"), ("time", "Time")]
    )


class WorkoutCreateForm(FlaskForm):
    name = StringField("name", [DataRequired()])
    exercises = FieldList(FormField(ExerciseCreateForm), min_entries=1)


class WorkoutEditForm(FlaskForm):
    name = StringField("name", [DataRequired()])
    exercises = FieldList(FormField(ExerciseEditForm), min_entries=1)

#  /workout/record


class SetForm(Form):
    lbs = FloatField("lbs", [Optional()])
    units = IntegerField("units", [Optional()])


class ExerciseRecordForm(Form):
    id = HiddenField("id", [DataRequired()])
    sets = FieldList(FormField(SetForm))


class WorkoutRecordForm(FlaskForm):
    class Meta:
        csrf = True
        csrf_class = SessionCSRF
        csrf_secret = urandom(16)
        csrf_time_limit = timedelta(minutes=120)
    exercises = FieldList(FormField(ExerciseRecordForm))