Coverage for polls/views.py: 72.73%

1from django.contrib.auth.mixins import LoginRequiredMixin (empty)

2from django.db.models import F (empty)

3from django.http import HttpResponseRedirect (empty)

4from django.shortcuts import get_object_or_404, render (empty)

5from django.urls import reverse (empty)

6from django.utils import timezone (empty)

7from django.views.generic import DetailView, ListView (empty)

8 

9from .models import Choice, Question (empty)

10 

11 

12class PollListView(LoginRequiredMixin, ListView): (empty)

13 template_name = "polls/poll_list.html" (empty)

14 context_object_name = "latest_question_list" (empty)

15 

16 def get_queryset(self): (empty)

17 """ 

18 Return the last five published questions (not including 

19 those set to be published in the future. 

20 """ 

21 return Question.objects.filter(pub_date__lte=timezone.now()).order_by( 5 ctx1bcdef

22 "-pub_date" 

23 )[:5] 

24 

25 

26class PollDetailView(LoginRequiredMixin, DetailView): (empty)

27 model = Question (empty)

28 template_name = "polls/poll_detail.html" (empty)

29 

30 def get_queryset(self): (empty)

31 """ 

32 Excludes any questions that aren't published yet. 

33 """ 

34 return Question.objects.filter(pub_date__lte=timezone.now()) 2 ctx1gh

35 

36 

37class PollResultsView(DetailView, LoginRequiredMixin): (empty)

38 model = Question (empty)

39 template_name = "polls/poll_results.html" (empty)

40 

41 def get_queryset(self): (empty)

42 """ 

43 Excludes any questions that aren't published yet. 

44 """ 

45 return Question.objects.filter(pub_date__lte=timezone.now()) 2 ctx1ij

46 

47 

48def vote(request, question_id): (empty)

49 question = get_object_or_404(Question, pk=question_id) 

50 try: 

51 selected_choice = question.choice_set.get(pk=request.POST["choice"]) 

52 except (KeyError, Choice.DoesNotExist): 

53 # Redisplay the question voting form. 

54 return render( 

55 request, 

56 "polls/poll_detail.html", 

57 { 

58 "question": question, 

59 "error_message": "You didn't select a choice.", 

60 }, 

61 ) 

62 else: 

63 selected_choice.votes = F("votes") + 1 

64 selected_choice.save() 

65 """ 

66 Always return an HttpResponseRedirect after successfully 

67 dealing with POST data. This prevents data from being posted 

68 twice if a user hits the Back Button. 

69 """ 

70 return HttpResponseRedirect(reverse("poll_results", args=(question.id,)))