from flask_wtf import Form
from wtforms import SelectField, SubmitField
from wtforms.validators import DataRequired


class DateForm(Form):
    year = SelectField('Year', validators=[DataRequired()])
    month = SelectField('Month',
                        choices=[
                            ('{:02d}'.format(i), str(i)) for i in range(1, 13)
                        ],
                        validators=[DataRequired()])

    submit = SubmitField('Submit')