Mercurial > public > sg101

annotate polls/views.py @ 1198:3a03c2b2df05

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix url return value bug.
author Brian Neal <bgneal@gmail.com>
date Sun, 07 May 2023 19:27:19 -0500
parents f7554fb88727
children
rev   line source
bgneal@439 1 """
bgneal@439 2 Views for the polls application.
gremmie@1 3
bgneal@439 4 """
bgneal@447 5 import datetime
bgneal@447 6
bgneal@439 7 from django.shortcuts import render
gremmie@1 8 from django.contrib.auth.decorators import login_required
gremmie@1 9 from django.shortcuts import get_object_or_404
gremmie@1 10 from django.http import Http404
gremmie@1 11 from django.http import HttpResponseRedirect
gremmie@1 12 from django.core.urlresolvers import reverse
bgneal@439 13 from django.views.decorators.http import require_POST
bgneal@439 14 from django.db.models import F
gremmie@1 15
gremmie@1 16 from polls.models import Poll
gremmie@1 17 from polls.models import Choice
gremmie@1 18 from polls.forms import VoteForm
gremmie@1 19
gremmie@1 20 #######################################################################
gremmie@1 21
bgneal@439 22 def get_user_choice(user, poll):
bgneal@439 23 """
bgneal@447 24 Return the Choice object the given user voted for from the given poll,
bgneal@447 25 or None if no vote has been recorded (or the user is not authenticated.
bgneal@439 26
bgneal@439 27 """
bgneal@439 28 user_choice = None
bgneal@439 29 if user.is_authenticated():
bgneal@439 30 user_choices = user.choice_set.filter(poll=poll)
bgneal@439 31 if user_choices:
bgneal@439 32 user_choice = user_choices[0]
bgneal@439 33
bgneal@439 34 return user_choice
bgneal@439 35
bgneal@439 36 #######################################################################
bgneal@439 37
gremmie@1 38 def poll_index(request):
bgneal@1090 39
bgneal@1090 40 # Do some stuff manually to avoid cascading hits to the database
bgneal@1090 41 now = datetime.datetime.now()
bgneal@1090 42 qs = Poll.objects.filter(is_enabled=True)
bgneal@1090 43 poll_dict = {}
bgneal@1090 44 current_polls = []
bgneal@1090 45 old_polls = []
bgneal@1090 46 for poll in qs:
bgneal@1090 47 poll.total_votes_ = 0
bgneal@1090 48 poll_dict[poll.pk] = poll
bgneal@1090 49 if (poll.start_date <= now and
bgneal@1090 50 (poll.end_date is None or poll.end_date >= now)):
bgneal@1090 51 current_polls.append(poll)
bgneal@1090 52 elif (poll.start_date < now and
bgneal@1090 53 (poll.end_date is not None and poll.end_date < now)):
bgneal@1090 54 old_polls.append(poll)
bgneal@1090 55
bgneal@1090 56 for choice in Choice.objects.iterator():
bgneal@1090 57 poll = poll_dict.get(choice.poll_id)
bgneal@1090 58 if poll:
bgneal@1090 59 poll.total_votes_ += choice.votes
bgneal@1090 60
bgneal@439 61 return render(request, 'polls/index.html', {
bgneal@439 62 'current_polls': current_polls,
bgneal@439 63 'old_polls': old_polls,
bgneal@1091 64 'V3_DESIGN': True,
bgneal@439 65 })
gremmie@1 66
gremmie@1 67 #######################################################################
gremmie@1 68
gremmie@1 69 def poll_detail(request, poll_id):
bgneal@439 70 poll = get_object_or_404(Poll, pk=poll_id)
bgneal@447 71 if not poll.is_enabled or poll.start_date > datetime.datetime.now():
bgneal@439 72 raise Http404
gremmie@1 73
bgneal@447 74 total_votes, choices = poll.results()
bgneal@447 75
bgneal@447 76 return render(request, 'polls/poll_detail.html', {
bgneal@439 77 'poll': poll,
bgneal@447 78 'total_votes': total_votes,
bgneal@447 79 'choices': choices,
bgneal@439 80 'user_choice': get_user_choice(request.user, poll),
bgneal@1091 81 'V3_DESIGN': True,
bgneal@439 82 })
gremmie@1 83
gremmie@1 84 #######################################################################
gremmie@1 85
gremmie@1 86 @login_required
gremmie@1 87 def poll_vote(request, poll_id):
bgneal@439 88 poll = get_object_or_404(Poll, pk=poll_id)
bgneal@439 89 if not poll.is_enabled:
bgneal@439 90 raise Http404
bgneal@439 91 if not poll.is_open():
bgneal@447 92 return HttpResponseRedirect(reverse('polls-detail',
bgneal@447 93 kwargs={'poll_id': poll_id}))
gremmie@1 94
bgneal@439 95 user_choice = get_user_choice(request.user, poll)
gremmie@1 96
bgneal@439 97 if request.method == "POST":
bgneal@447 98 vote_form = VoteForm(poll, request.POST, user=request.user,
bgneal@447 99 user_choice=user_choice)
bgneal@439 100 if vote_form.is_valid():
bgneal@439 101 vote_form.save()
bgneal@447 102 return HttpResponseRedirect(reverse('polls-detail',
bgneal@447 103 kwargs={'poll_id': poll_id}))
bgneal@439 104 else:
bgneal@439 105 vote_form = VoteForm(poll)
bgneal@439 106
bgneal@439 107 return render(request, 'polls/poll_vote.html', {
bgneal@439 108 'poll': poll,
bgneal@439 109 'vote_form': vote_form,
bgneal@439 110 'user_choice': user_choice,
bgneal@1091 111 'V3_DESIGN': True,
bgneal@439 112 })
gremmie@1 113
gremmie@1 114 #######################################################################
gremmie@1 115
bgneal@439 116 @require_POST
bgneal@439 117 @login_required
bgneal@439 118 def poll_delete_vote(request):
bgneal@439 119 poll = get_object_or_404(Poll, pk=request.POST.get('poll_id'))
bgneal@439 120 user_choice = get_user_choice(request.user, poll)
bgneal@439 121 if user_choice:
bgneal@439 122 Choice.objects.filter(id=user_choice.id).update(votes=F('votes') - 1)
bgneal@439 123 user_choice.voters.remove(request.user)
bgneal@439 124
bgneal@447 125 return HttpResponseRedirect(reverse('polls-detail', kwargs={'poll_id': poll.id}))