1 files changed, 16 insertions, 1 deletions
diff --git a/forms.py b/forms.py
index be0171a..ba427cd 100644
--- a/forms.py
+++ b/forms.py
@@ -1,13 +1,28 @@
 from flask_wtf import FlaskForm
-from wtforms import StringField, TextAreaField, SubmitField, PasswordField, BooleanField
+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"""