Code layout change.

[wolnelektury.git] / src / polls / templatetags / polls_tags.py

diff --git a/src/polls/templatetags/polls_tags.py b/src/polls/templatetags/polls_tags.py
new file mode 100644 (file)
index 0000000..38619c1
--- /dev/null
+++ b/src/polls/templatetags/polls_tags.py
@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from django import template
+from ..forms import PollForm
+
+
+register = template.Library()
+
+@register.inclusion_tag('polls/tags/poll.html', takes_context=True)
+def poll(context, poll, show_results=True, redirect_to=''):
+    form = None
+    voted_already = poll.voted(context.get('request').session)
+    if not voted_already:
+        form = PollForm(poll=poll, initial=dict(redirect_to=redirect_to))
+    return dict(poll=poll,
+                form=form,
+                voted_already=voted_already,
+                vote_count=poll.vote_count,
+                show_results=show_results,
+                request=context.get('request'),
+           )