Code layout change.

[wolnelektury.git] / src / polls / forms.py

diff --git a/src/polls/forms.py b/src/polls/forms.py
new file mode 100644 (file)
index 0000000..20d53fe
--- /dev/null
+++ b/src/polls/forms.py
@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-\r
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.\r
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.\r
+#\r
+from django import forms\r
+\r
+\r
+class PollForm(forms.Form):\r
+    vote = forms.ChoiceField(widget=forms.RadioSelect)\r
+\r
+    def __init__(self, *args, **kwargs):
+        poll = kwargs.pop('poll', None)\r
+        super(PollForm, self).__init__(*args, **kwargs)\r
+        self.fields['vote'].choices = [(item.id, item.content) for item in poll.items.all()]\r