Code layout change.

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

diff --git a/src/polls/views.py b/src/polls/views.py
new file mode 100644 (file)
index 0000000..79540c6
--- /dev/null
+++ b/src/polls/views.py
@@ -0,0 +1,37 @@
+# -*- 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.core.urlresolvers import reverse
+from django.shortcuts import get_object_or_404, redirect, render_to_response
+from django.template import RequestContext
+from django.views.decorators import cache
+from django.views.decorators.http import require_http_methods
+
+from .models import Poll, PollItem
+from .forms import PollForm
+
+
+@cache.never_cache
+@require_http_methods(['GET', 'POST'])
+def poll(request, slug):
+
+    poll = get_object_or_404(Poll, slug=slug, open=True)
+
+    if request.method == 'POST':
+        redirect_to = reverse('poll', args = [slug])
+        form = PollForm(request.POST, poll = poll)
+        if form.is_valid():
+            if not poll.voted(request.session):
+                try:
+                    poll_item = PollItem.objects.filter(pk=form.cleaned_data['vote'], poll=poll).get()
+                except PollItem.DoesNotExist:
+                    pass
+                else:
+                    poll_item.vote(request.session)
+        return redirect(redirect_to)
+    elif request.method == 'GET':
+        context = RequestContext(request)
+        context['poll'] = poll
+        context['voted_already'] = poll.voted(request.session)
+        return render_to_response('polls/poll.html', context)