diff --git a/atst/forms/new_member.py b/atst/forms/new_member.py index b7358d00..5b53c839 100644 --- a/atst/forms/new_member.py +++ b/atst/forms/new_member.py @@ -1,4 +1,4 @@ -from flask_wtf import Form +from flask_wtf import FlaskForm from wtforms.fields import StringField from wtforms.fields.html5 import EmailField from wtforms.validators import Required, Email, Length @@ -9,7 +9,7 @@ from atst.forms.fields import SelectField from .data import WORKSPACE_ROLES -class NewMemberForm(Form): +class NewMemberForm(FlaskForm): first_name = StringField(label="First Name", validators=[Required()]) last_name = StringField(label="Last Name", validators=[Required()]) diff --git a/atst/forms/new_project.py b/atst/forms/new_project.py index b6007364..00b01325 100644 --- a/atst/forms/new_project.py +++ b/atst/forms/new_project.py @@ -1,10 +1,10 @@ -from flask_wtf import Form +from flask_wtf import FlaskForm from wtforms.fields import StringField, TextAreaField, FieldList from wtforms.validators import Required from atst.forms.validators import ListItemRequired, ListItemsUnique -class NewProjectForm(Form): +class NewProjectForm(FlaskForm): EMPTY_ENVIRONMENT_NAMES = ["", None] @@ -20,7 +20,7 @@ class NewProjectForm(Form): @property def data(self): - _data = super(Form, self).data + _data = super(FlaskForm, self).data _data["environment_names"] = [ n for n in _data["environment_names"]