80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
from django.contrib.messages.views import SuccessMessageMixin
|
|
from django.db.models import Count, Prefetch
|
|
from django.http import HttpResponseRedirect
|
|
from django.urls import reverse
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.views.generic import CreateView, DetailView, UpdateView
|
|
|
|
from .forms import OptionFormSet
|
|
from .models import Election, Option, Question
|
|
|
|
# TODO: access control *everywhere*
|
|
|
|
|
|
class ElectionCreateView(SuccessMessageMixin, CreateView):
|
|
model = Election
|
|
template_name = "elections/election_create.html"
|
|
fields = ["name", "description", "start_time", "end_time"]
|
|
success_message = _("Élection crée avec succès !")
|
|
|
|
|
|
class ElectionUpdateView(SuccessMessageMixin, UpdateView):
|
|
model = Election
|
|
fields = ["name", "description", "start_time", "end_time"]
|
|
success_message = _("Élection modifiée avec succès !")
|
|
|
|
def get_queryset(self):
|
|
# On ne peut plus modifier une élection déjà comptée
|
|
return super().get_queryset().filter(tallied=False, archived=False)
|
|
|
|
|
|
class ElectionView(DetailView):
|
|
model = Election
|
|
template_name = "elections/election.html"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
if self.object.tallied:
|
|
options_qs = Option.objects.annotate(nb_votes=Count("voters"))
|
|
questions = self.election.question.prefetch_related(
|
|
Prefetch("options", queryset=options_qs)
|
|
)
|
|
context["questions"] = questions
|
|
|
|
return context
|
|
|
|
def get_queryset(self):
|
|
return super().get_queryset().filter(archived=False)
|
|
|
|
|
|
class VoteView(DetailView):
|
|
model = Question
|
|
template_name = "elections/vote.html"
|
|
|
|
def get_queryset(self):
|
|
# On ne peut voter que si l'élection n'a pas été comptée
|
|
return (
|
|
super()
|
|
.get_queryset()
|
|
.filter(election__tallied=False, election__archived=False)
|
|
.select_related("election")
|
|
)
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
vote_form = OptionFormSet(instance=self.object)
|
|
|
|
return self.render_to_response(self.get_context_data(vote_form=vote_form))
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
vote_form = OptionFormSet(self.request.POST, instance=self.object)
|
|
|
|
# We record the cast votes
|
|
for v in vote_form:
|
|
v.record_vote(self.request.user)
|
|
|
|
return HttpResponseRedirect(
|
|
reverse("election.view", args=[self.object.election.pk])
|
|
+ f"#q_{self.object.pk}"
|
|
)
|