flask_photo_scaling_app/app/forms.py

#!/usr/bin/env python3

from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, BooleanField, SubmitField
from wtforms.validators import DataRequired, Email, Optional, Regexp, ValidationError, EqualTo, Length
from flask_wtf.file import FileField, FileAllowed, FileRequired
from app.models import Contributor, EmailWhiteList
from zxcvbn import zxcvbn


class ConfirmPhotoDelete(FlaskForm):
    submit = SubmitField('Delete')


class DisableTotp(FlaskForm):
    submit = SubmitField('Disable 2FA')


class GetTotp(FlaskForm):
    totp_code = StringField('6-Digit Code?', validators=[DataRequired(), Length(min=6, max=6, message="6 Digits")], render_kw={'autofocus': True})
    submit = SubmitField('OK')


class ConfirmTotp(FlaskForm):
    totp_code = StringField('6-Digit Code?', validators=[DataRequired(), Length(min=6, max=6, message="Rescan And Try Again")], render_kw={'autofocus': True})
    submit = SubmitField('Enable 2FA')


class EditProfile(FlaskForm):
    username = StringField('Username', validators=[DataRequired(), Regexp('^[a-zA-Z0-9]+$', message='letters and digits only (no spaces)')], render_kw={'autofocus': True})
    email = StringField('Email', validators=[Optional(), Email()])
    password = PasswordField('Confirm Password', validators=[DataRequired()])
    submit = SubmitField('Update Name/Email')

    def validate_username(self, username):
        from flask_login import current_user
        if username.data != current_user.name:
            user = Contributor.query.filter_by(name=username.data).first()
            if user is not None:
                raise ValidationError('Please pick a different username.')

    def validate_email(self, email):
        from flask_login import current_user
        if email.data != current_user.email:
            user = Contributor.query.filter_by(email=email.data).first()
            if user is not None:
                raise ValidationError('Please pick a different email address.')


class LoginForm(FlaskForm):
    username = StringField('Username', validators=[Optional()], render_kw={'autofocus': True})
    email = StringField('Email', validators=[Optional(), Email()])
    password = PasswordField('Password', validators=[DataRequired()])
    remember_me = BooleanField('Remember Me')
    submit = SubmitField('Sign In')


class ResetPasswordForm(FlaskForm):
    password = PasswordField('Password', validators=[DataRequired(), Length(min=15, )], render_kw={'autofocus': True})
    password2 = PasswordField('Repeat Password', validators=[DataRequired(), EqualTo('password')])
    submit = SubmitField('Request Password Reset')

    def validate_password(self, password):
        if zxcvbn(password.data)['score'] < 3:
            raise ValidationError('needs to be stronger')


class RegistrationForm(FlaskForm):
    username = StringField('Username', validators=[DataRequired(), Regexp('^[a-zA-Z0-9]+$', message='letters and digits only (no spaces)')], render_kw={'autofocus': True})
    email = StringField('Email', validators=[DataRequired(), Email()])
    password = PasswordField('Password', validators=[DataRequired(), Length(min=15, )])
    password2 = PasswordField('Repeat Password', validators=[DataRequired(), EqualTo('password')])
    submit = SubmitField('Register')

    def validate_password(self, password):
        if zxcvbn(password.data)['score'] < 3:
            raise ValidationError('needs to be stronger')

    def validate_username(self, username):
        user = Contributor.query.filter_by(name=username.data).first()
        if user is not None:
            raise ValidationError('Please use a different username.')

    def validate_email(self, email):
        white_listed_user = EmailWhiteList.query.filter_by(email=email.data).first()
        if white_listed_user is None:
            raise ValidationError('This email address is not authorized.')
        user = Contributor.query.filter_by(email=email.data).first()
        if user is not None:
            raise ValidationError('Please use a different email address.')


class ResetPasswordRequestForm(FlaskForm):
    email = StringField('Email', validators=[DataRequired(), Email()], render_kw={'autofocus': True})
    submit = SubmitField('Request Password Reset')


class ChangePassword(FlaskForm):
    password = PasswordField('Confirm Password', validators=[DataRequired()], render_kw={'autofocus': True})
    new_password = PasswordField('New Password', validators=[DataRequired(), Length(min=15, )])
    new_password2 = PasswordField('Repeat New Password', validators=[DataRequired(), EqualTo('new_password')])
    submit = SubmitField('Save')

    def validate_password(self, password):
        if zxcvbn(password.data)['score'] < 3:
            raise ValidationError('needs to be stronger')


class UploadPhotoForm(FlaskForm):
    image = FileField('Upload (<=3M)', validators=[
        FileRequired(),
        FileAllowed(['jpg', 'png'], 'The file format should be .jpg or .png.')
    ])
    submit = SubmitField()