from wtforms import Form, TextField, TextAreaField, PasswordField, SubmitField, validators
from models import db, Users


class RegistrationForm(Form):
    firstname = TextField("First name",  [validators.Required("Please enter your first name.")])
    lastname = TextField("Last name",  [validators.Required("Please enter your last name.")])
    username = TextField('Username', [validators.Required("Please enter your email.")])
    email = TextField('Email Address', [validators.Email("Please enter your email address.")])
    password = PasswordField('Password')
    confirm = PasswordField('Repeat Password')

    def __init__(self, *args, **kwargs):

        Form.__init__(self, *args, **kwargs)

    def validate(self):
        import pdb
        pdb.set_trace()
        if not Form.validate(self):
            return False

        user = Users.query.filter_by(email=self.email.data.lower()).first()
        if user:
            self.email.errors.append("That email is already taken")
            return False
        else:
            return True