summaryrefslogtreecommitdiffstats
path: root/polls/views.py
diff options
context:
space:
mode:
Diffstat (limited to 'polls/views.py')
-rwxr-xr-xpolls/views.py38
1 files changed, 38 insertions, 0 deletions
diff --git a/polls/views.py b/polls/views.py
new file mode 100755
index 0000000..18bf3ee
--- /dev/null
+++ b/polls/views.py
@@ -0,0 +1,38 @@
+# Create your views here.
+
+from django.shortcuts import render_to_response, get_object_or_404
+from django.core.urlresolvers import reverse
+from infopoint.polls.models import Poll
+from django.http import Http404, HttpResponseRedirect
+
+def index(request):
+ object_list = Poll.objects.all().order_by('-pub_date')[:5]
+ return render_to_response('polls/index.html', {'object_list': object_list})
+
+def detail(request, poll_id):
+ p = get_object_or_404(Poll, pk=poll_id)
+ return render_to_response('polls/detail.html', {'poll': p})
+
+
+def vote(request, poll_id):
+ p = get_object_or_404(Poll, pk=poll_id)
+ try:
+ selected_choice = p.choice_set.get(pk=request.POST['choice'])
+ except (KeyError, Choice.DoesNotExist):
+ # Redisplay the poll voting form.
+ return render_to_response('polls/detail.html', {
+ 'poll': p,
+ 'error_message': "You didn't select a choice.",
+ })
+ else:
+ selected_choice.votes += 1
+ selected_choice.save()
+ # Always return an HttpResponseRedirect after successfully dealing
+ # with POST data. This prevents data from being posted twice if a
+ # user hits the Back button.
+ return HttpResponseRedirect(reverse('infopoint.polls.views.results', args=(p.id,)))
+
+def results(request, poll_id):
+ p = get_object_or_404(Poll, pk=poll_id)
+ return render_to_response('polls/results.html', {'poll': p})
+