Rensat i koden

app/forms.py

@@ -1,7 +1,8 @@
 from flask_wtf import FlaskForm
-from wtforms import StringField, BooleanField, TextAreaField, SelectField, SelectMultipleField, SubmitField
+from wtforms import StringField, BooleanField, TextAreaField, SubmitField
 from wtforms import widgets
 from wtforms.validators import DataRequired
+from wtforms.ext.sqlalchemy.fields import QuerySelectField, QuerySelectMultipleField
 
 class PageForm(FlaskForm):
 
@@ -9,6 +10,9 @@ class PageForm(FlaskForm):
     title = StringField('Title', validators=[DataRequired()])
     description = TextAreaField('Beskrivning')
     endpoint = BooleanField('Produkt')
-    thumbnail = SelectField('Bild')
-    photos = SelectMultipleField('Galleri', option_widget=widgets.CheckboxInput(), widget=widgets.ListWidget(prefix_label=False))
+    thumbnail = QuerySelectField('Bild', get_label='alt')
+    photos = QuerySelectMultipleField('Galleri',
+            option_widget=widgets.CheckboxInput(),
+            widget=widgets.ListWidget(prefix_label=False),
+            get_label = lambda p: '<img src="/thumbnail/{}">'.format(p.id))
     submit = SubmitField('Ok')