Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from flask_wtf import FlaskForm
- from wtforms import PasswordField, StringField, SubmitField, ValidationError
- from wtforms.validators import DataRequired, Email, EqualTo
- from ..models import Employee
- class RegistrationForm(FlaskForm):
- """
- Form for users to create new account
- """
- email = StringField('Email', validators = [DataRequired(), Email])
- username = StringField('Username', validators = [DataRequired()])
- first_name = StringField('First Name', validators = [DataRequired()])
- last_name = StringField('Last Name', validators = [DataRequired()])
- password = PasswordField('Password', validators = [DataRequired(), EqualTo('confirm_password')])
- confirm_password = PasswordField('Confirm Password')
- submit = SubmitField('Register')
- def validate_email(self, field):
- if Employee.query.filter_by(email=field.data).first():
- raise ValidationError('Email is already in use.')
- def validate_username(self, field):
- if Employee.query.filter_by(username=field.data).first():
- raise ValiationError("Username is already in use.")
- class LoginForm(FlaskForm):
- """
- Form for users to login
- """
- email = StringField('Email', validators = [DataRequired(), Email()])
- password = PasswordField('Password', validators = [DataRequired()])
- submit = SubmitField('Login')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement