from flask_wtf import FlaskForm from wtforms import ( StringField, TextAreaField, SubmitField, PasswordField, BooleanField, SelectField, ) from wtforms.validators import DataRequired class PostForm(FlaskForm): title = StringField("Title", validators=[DataRequired()]) content = TextAreaField("Content", validators=[DataRequired()]) faccets = SelectField("Faccet", coerce=int) submit = SubmitField("Toot!") def __init__(self, *args, **kwargs): super(PostForm, self).__init__(*args, **kwargs) # Populate the select field with profile from the database self.faccets.choices = [(0, "Please select a profile")] + [ (u.id, u.name) for u in kwargs["faccets"] ] class LoginForm(FlaskForm): """Login Form""" username = StringField("Username", validators=[DataRequired()]) password = PasswordField("Password", validators=[DataRequired()]) remember_me = BooleanField("Remember Me") submit = SubmitField("Sign In")