from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField, PasswordField
from flask_wtf.file import FileAllowed, FileField
from wtforms.validators import DataRequired


class LoginForm(FlaskForm):
    username = StringField('Username',validators=[DataRequired()])
    password = PasswordField('Password',validators=[DataRequired()])
    submit = SubmitField('Login')


class UpdateForm(FlaskForm):
    file = FileField('Update File', validators=[DataRequired(), FileAllowed(['pdf'])])
    submit = SubmitField('Update')