panikdb/poll/views.py

   1 from django.contrib.auth.decorators import login_required
   2 from django.shortcuts import redirect
   3 from django.views.generic.base import TemplateView
   4 from django.contrib import messages
   5
   6 from .models import Vote
   7
   8 class VoteView(TemplateView):
   9     template_name = 'vote.html'
  10
  11     def dispatch(self, request, *args, **kwargs):
  12         return redirect('home')
  13         try:
  14             vote = Vote.objects.get(user=request.user)
  15             messages.error(request, 'Non !')
  16             return redirect('home')
  17         except Vote.DoesNotExist:
  18             pass
  19         return super(VoteView, self).dispatch(request, *args, **kwargs)
  20
  21     def post(self, request):
  22         choice = request.POST['choice']
  23         comment = request.POST['comment']
  24
  25         vote = Vote(
  26                 user=request.user,
  27                 choice=request.POST['choice'],
  28                 comment=request.POST['comment'])
  29         vote.save()
  30         messages.success(request, 'Merci !')
  31         return redirect('home')
  32
  33 vote = login_required(VoteView.as_view())