# -*- mode: python; -*-


from flask_wtf import FlaskForm
from wtforms import (
    SubmitField,
    HiddenField,
    StringField,
    PasswordField,
    BooleanField,
)
from wtforms.validators import (
    InputRequired,
    Length,
    ValidationError,
)


class LoginForm(FlaskForm):
    username = StringField("Username", validators=[InputRequired()])
    password = PasswordField("Password", validators=[InputRequired()])
    remember = BooleanField("Remember")
    submit = SubmitField("Login")


class RegisterForm(LoginForm):
    def validate_invite_code(self, field):
        if field.data != "mdltesters2022":
            raise ValidationError("Invitation code does not match")

    invitation_code = StringField("Invitation code", validators=[InputRequired()])
    name_first = StringField("First name", validators=[InputRequired()])
    name_last = StringField("Last name", validators=[InputRequired()])
    submit = SubmitField("Register")