restructuring for poetry-slam

src/ttfrog/webserver/forms.py

@@ -0,0 +1,21 @@
+from wtforms.fields import SelectField, SelectMultipleField
+
+from ttfrog.db.manager import db
+
+
+class DeferredSelectMultipleField(SelectMultipleField):
+    def __init__(self, *args, model=None, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.choices = [(rec.id, rec.name) for rec in db.query(model).all()]
+
+
+class DeferredSelectField(SelectField):
+    def __init__(self, *args, model=None, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.choices = [(rec.id, getattr(rec, "name", str(rec))) for rec in db.query(model).all()]
+
+
+class NullableDeferredSelectField(DeferredSelectField):
+    def __init__(self, *args, model=None, label="---", **kwargs):
+        super().__init__(*args, model=model, **kwargs)
+        self.choices = [(0, label)] + self.choices