/
forms.py
27 lines (19 loc) · 887 Bytes
/
forms.py
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
from flask_wtf import FlaskForm
from wtforms.validators import DataRequired, ValidationError
from wtforms import DateField, StringField
from models import Planning
class AddPlanning(FlaskForm):
"""Add planning to database"""
planning_name = StringField('Planning name',
validators=[DataRequired()])
start_date = DateField('begin date DD/MM/YY',
format="%d/%m/%Y",
validators=[DataRequired()])
end_date = DateField('end date DD/MM/YY',
format="%d/%m/%Y",
validators=[DataRequired()])
class AddTask(FlaskForm):
"""Add Task to your planning"""
def planning_exists(form, field):
if Planning.select().where(Planning.planning_name == field.data).exists():
raise ValidationError('User with that name already exists.')