summary refs log tree commit diff
path: root/forms.py
blob: ba427cd193ed7eb9952405cd00e459415af75c0c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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")