Mercurial > public > sg101

annotate gpp/polls/views.py @ 446:72ef6e809f79

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
For #218, put back the logic that updates the topic last visit after you make a post.
author Brian Neal <bgneal@gmail.com>
date Mon, 13 Jun 2011 02:19:13 +0000
parents 1f139de929c4
children 8f46ba2f1b81
rev   line source
bgneal@439 1 """
bgneal@439 2 Views for the polls application.
gremmie@1 3
bgneal@439 4 """
bgneal@439 5 from django.shortcuts import render
gremmie@1 6 from django.contrib.auth.decorators import login_required
gremmie@1 7 from django.shortcuts import get_object_or_404
gremmie@1 8 from django.http import Http404
gremmie@1 9 from django.http import HttpResponseRedirect
gremmie@1 10 from django.core.urlresolvers import reverse
bgneal@439 11 from django.views.decorators.http import require_POST
bgneal@439 12 from django.db.models import F
gremmie@1 13
gremmie@1 14 from polls.models import Poll
gremmie@1 15 from polls.models import Choice
gremmie@1 16 from polls.forms import VoteForm
gremmie@1 17
gremmie@1 18 #######################################################################
gremmie@1 19
bgneal@439 20 def get_user_choice(user, poll):
bgneal@439 21 """
bgneal@439 22 Return the Choice object the give user voted for from the given poll,
bgneal@439 23 or None of no vote has been recorded (or the user is not authenticated.
bgneal@439 24
bgneal@439 25 """
bgneal@439 26 user_choice = None
bgneal@439 27 if user.is_authenticated():
bgneal@439 28 user_choices = user.choice_set.filter(poll=poll)
bgneal@439 29 if user_choices:
bgneal@439 30 user_choice = user_choices[0]
bgneal@439 31
bgneal@439 32 return user_choice
bgneal@439 33
bgneal@439 34 #######################################################################
bgneal@439 35
gremmie@1 36 def poll_index(request):
bgneal@439 37 current_polls = Poll.objects.get_current_polls()
bgneal@439 38 old_polls = Poll.objects.get_old_polls()
bgneal@439 39 return render(request, 'polls/index.html', {
bgneal@439 40 'current_polls': current_polls,
bgneal@439 41 'old_polls': old_polls,
bgneal@439 42 })
gremmie@1 43
gremmie@1 44 #######################################################################
gremmie@1 45
gremmie@1 46 def poll_detail(request, poll_id):
bgneal@439 47 poll = get_object_or_404(Poll, pk=poll_id)
bgneal@439 48 if not poll.is_enabled:
bgneal@439 49 raise Http404
gremmie@1 50
bgneal@439 51 return render(request, 'polls/poll.html', {
bgneal@439 52 'poll': poll,
bgneal@439 53 'user_choice': get_user_choice(request.user, poll),
bgneal@439 54 })
gremmie@1 55
gremmie@1 56 #######################################################################
gremmie@1 57
gremmie@1 58 @login_required
gremmie@1 59 def poll_vote(request, poll_id):
bgneal@439 60 poll = get_object_or_404(Poll, pk=poll_id)
bgneal@439 61 if not poll.is_enabled:
bgneal@439 62 raise Http404
bgneal@439 63 if not poll.is_open():
bgneal@439 64 return HttpResponseRedirect(reverse('polls-results', args=[poll_id]))
gremmie@1 65
bgneal@439 66 user_choice = get_user_choice(request.user, poll)
gremmie@1 67
bgneal@439 68 if request.method == "POST":
bgneal@439 69 vote_form = VoteForm(poll, request.POST, user=request.user, user_choice=user_choice)
bgneal@439 70 if vote_form.is_valid():
bgneal@439 71 vote_form.save()
bgneal@439 72 return HttpResponseRedirect(reverse('polls-results', args=[poll_id]))
bgneal@439 73 else:
bgneal@439 74 vote_form = VoteForm(poll)
bgneal@439 75
bgneal@439 76 return render(request, 'polls/poll_vote.html', {
bgneal@439 77 'poll': poll,
bgneal@439 78 'vote_form': vote_form,
bgneal@439 79 'user_choice': user_choice,
bgneal@439 80 })
gremmie@1 81
gremmie@1 82 #######################################################################
gremmie@1 83
gremmie@1 84 def poll_results(request, poll_id):
bgneal@439 85 poll = get_object_or_404(Poll, pk=poll_id)
bgneal@439 86 total_votes, choices = poll.results()
bgneal@439 87 return render(request, 'polls/poll_results.html', {
bgneal@439 88 'poll': poll,
bgneal@439 89 'total_votes': total_votes,
bgneal@439 90 'choices': choices,
bgneal@439 91 'user_choice': get_user_choice(request.user, poll),
bgneal@439 92 })
gremmie@1 93
gremmie@1 94 #######################################################################
bgneal@439 95
bgneal@439 96 @require_POST
bgneal@439 97 @login_required
bgneal@439 98 def poll_delete_vote(request):
bgneal@439 99 poll = get_object_or_404(Poll, pk=request.POST.get('poll_id'))
bgneal@439 100 user_choice = get_user_choice(request.user, poll)
bgneal@439 101 if user_choice:
bgneal@439 102 Choice.objects.filter(id=user_choice.id).update(votes=F('votes') - 1)
bgneal@439 103 user_choice.voters.remove(request.user)
bgneal@439 104
bgneal@439 105 return HttpResponseRedirect(reverse('polls-results', args=[poll.id]))