kadenios/elections/models.py

251 lines
7.5 KiB
Python
Raw Normal View History

from translated_fields import TranslatedFieldWithFallback
2020-12-20 18:50:38 +01:00
from django.conf import settings
from django.contrib.auth.models import AbstractUser
2021-04-09 03:31:43 +02:00
from django.db import models, transaction
from django.template.loader import render_to_string
2020-11-19 18:40:22 +01:00
from django.utils.translation import gettext_lazy as _
2020-11-19 17:29:43 +01:00
from shared.utils import choices_length
2021-03-19 16:08:02 +01:00
from .staticdefs import (
BALLOT_TYPE,
2021-03-19 16:08:02 +01:00
CAST_FUNCTIONS,
CONNECTION_METHODS,
QUESTION_TYPES,
TALLY_FUNCTIONS,
VALIDATE_FUNCTIONS,
2021-03-19 16:08:02 +01:00
)
from .utils import (
2021-04-17 00:23:33 +02:00
BallotsData,
CastFunctions,
ResultsData,
TallyFunctions,
ValidateFunctions,
)
2021-01-27 14:55:28 +01:00
# #############################################################################
# Models regarding an election
# #############################################################################
2020-11-19 18:40:22 +01:00
class Election(models.Model):
name = TranslatedFieldWithFallback(models.CharField(_("nom"), max_length=255))
2020-11-19 18:40:22 +01:00
short_name = models.SlugField(_("nom bref"), unique=True)
description = TranslatedFieldWithFallback(
models.TextField(_("description"), blank=True)
)
2020-11-19 18:40:22 +01:00
2020-12-19 15:04:04 +01:00
start_date = models.DateTimeField(_("date et heure de début"))
end_date = models.DateTimeField(_("date et heure de fin"))
2020-11-19 18:40:22 +01:00
visible = models.BooleanField(_("visible au public"), default=False)
vote_restrictions = TranslatedFieldWithFallback(
models.TextField(_("conditions de vote"), blank=True)
)
2020-12-20 18:50:38 +01:00
restricted = models.BooleanField(
_("restreint le vote à une liste de personnes"), default=True
)
sent_mail = models.BooleanField(
_("mail avec les identifiants envoyé"), default=False
)
2020-11-19 18:40:22 +01:00
created_by = models.ForeignKey(
2020-12-20 18:50:38 +01:00
settings.AUTH_USER_MODEL,
related_name="elections_created",
on_delete=models.SET_NULL,
blank=True,
null=True,
2020-11-19 18:40:22 +01:00
)
voters = models.ManyToManyField(
settings.AUTH_USER_MODEL,
related_name="cast_elections",
2021-03-18 14:49:48 +01:00
blank=True,
)
2020-11-19 18:40:22 +01:00
results_public = models.BooleanField(_("résultats publics"), default=False)
2020-11-20 14:55:31 +01:00
tallied = models.BooleanField(_("dépouillée"), default=False)
2020-11-19 18:40:22 +01:00
archived = models.BooleanField(_("archivée"), default=False)
2020-12-19 23:48:18 +01:00
class Meta:
2021-05-30 00:40:32 +02:00
permissions = [
("is_admin", _("Peut administrer des élections")),
]
2020-12-19 23:48:18 +01:00
ordering = ["-start_date", "-end_date"]
2020-11-19 18:40:22 +01:00
class Question(models.Model):
2020-11-20 14:55:31 +01:00
election = models.ForeignKey(
Election, related_name="questions", on_delete=models.CASCADE
)
2021-06-14 14:42:36 +02:00
text = TranslatedFieldWithFallback(
models.TextField(_("question"), blank=True, default="")
)
type = models.CharField(
_("type de question"),
choices=QUESTION_TYPES,
default="assentiment",
max_length=choices_length(QUESTION_TYPES),
)
# We cache the maximum number of votes for an option
max_votes = models.PositiveSmallIntegerField(
_("nombre maximal de votes reçus"), default=0
)
2020-11-19 18:40:22 +01:00
voters = models.ManyToManyField(
settings.AUTH_USER_MODEL,
related_name="cast_questions",
2021-03-18 14:49:48 +01:00
blank=True,
)
def is_form_valid(self, vote_form):
validate_function = getattr(ValidateFunctions, VALIDATE_FUNCTIONS[self.type])
return vote_form.is_valid() and validate_function(vote_form)
2021-04-09 03:31:43 +02:00
@transaction.atomic
2021-03-19 14:25:13 +01:00
def cast_ballot(self, user, vote_form):
cast_function = getattr(CastFunctions, CAST_FUNCTIONS[self.type])
cast_function(user, vote_form)
2021-04-09 03:31:43 +02:00
@transaction.atomic
2021-03-19 16:08:02 +01:00
def tally(self):
tally_function = getattr(TallyFunctions, TALLY_FUNCTIONS[self.type])
tally_function(self)
@property
def results(self):
return render_to_string(
f"elections/results/{self.vote_type}_export.txt", {"question": self}
)
def get_formset(self):
from .forms import BallotFormset # Avoid circular imports
return getattr(BallotFormset, BALLOT_TYPE[self.type])
def get_results_data(self):
results_function = getattr(ResultsData, BALLOT_TYPE[self.type])
return results_function(self)
2021-04-17 00:23:33 +02:00
def display_ballots(self):
display_function = getattr(BallotsData, BALLOT_TYPE[self.type])
return display_function(self)
@property
def vote_type(self):
return BALLOT_TYPE[self.type]
def __str__(self):
return self.text
2020-12-19 23:48:18 +01:00
class Meta:
ordering = ["id"]
2020-11-19 18:40:22 +01:00
class Option(models.Model):
2020-11-20 14:55:31 +01:00
question = models.ForeignKey(
Question, related_name="options", on_delete=models.CASCADE
)
text = TranslatedFieldWithFallback(models.TextField(_("texte"), blank=False))
2021-04-20 10:04:16 +02:00
abbreviation = models.CharField(_("abréviation"), max_length=3, blank=True)
winner = models.BooleanField(_("option gagnante"), default=False)
2020-11-20 14:55:31 +01:00
voters = models.ManyToManyField(
2020-12-20 18:50:38 +01:00
settings.AUTH_USER_MODEL,
2020-11-20 14:55:31 +01:00
related_name="votes",
through="Vote",
2021-03-18 14:49:48 +01:00
blank=True,
2020-11-20 17:45:15 +01:00
)
# For now, we store the amount of votes received after the election is tallied
nb_votes = models.PositiveSmallIntegerField(_("nombre de votes reçus"), default=0)
2020-12-19 23:48:18 +01:00
def save(self, *args, **kwargs):
# On enlève les espaces et on passe tout en majuscules
self.abbreviation = "".join(self.abbreviation.upper().split())
super().save(*args, **kwargs)
def __str__(self):
if self.abbreviation:
return self.abbreviation + " - " + self.text
return self.text
2020-12-19 23:48:18 +01:00
class Meta:
ordering = ["id"]
2020-12-20 18:50:38 +01:00
class Vote(models.Model):
option = models.ForeignKey(Option, on_delete=models.CASCADE)
user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
class Meta:
ordering = ["option"]
class Rank(models.Model):
vote = models.OneToOneField(Vote, on_delete=models.CASCADE)
rank = models.PositiveSmallIntegerField(_("rang de l'option"))
class Meta:
ordering = ["vote"]
class Duel(models.Model):
question = models.ForeignKey(
Question, related_name="duels", on_delete=models.CASCADE
)
winner = models.ForeignKey(
Option, related_name="contests_won", on_delete=models.CASCADE
)
loser = models.ForeignKey(
Option, related_name="contests_lost", on_delete=models.CASCADE
)
amount = models.PositiveSmallIntegerField(_("votes supplémentaires"))
2020-12-20 18:50:38 +01:00
# #############################################################################
# Modification of the base User Model
# #############################################################################
class User(AbstractUser):
election = models.ForeignKey(
Election,
related_name="registered_voters",
null=True,
blank=True,
on_delete=models.CASCADE,
)
2020-12-23 18:04:39 +01:00
full_name = models.CharField(_("Nom et Prénom"), max_length=150, blank=True)
2020-12-20 18:50:38 +01:00
@property
def base_username(self):
return "__".join(self.username.split("__")[1:])
2020-12-21 00:07:07 +01:00
def can_vote(self, request, election):
2020-12-20 18:50:38 +01:00
# Si c'est un·e utilisateur·ice CAS, iel peut voter dans les élections
# ouvertes à tou·te·s
if self.election is None:
# If the user is connected via CAS, request.session["CASCONNECTED"] is set
# to True by authens
return not election.restricted and request.session.get("CASCONNECTED")
2020-12-20 18:50:38 +01:00
# Pour les élections restreintes, il faut y être associé
return election.restricted and (self.election == election)
2020-12-21 00:07:07 +01:00
def get_prefix(self):
return self.username.split("__")[0]
@property
2020-12-21 00:07:07 +01:00
def connection_method(self):
2021-01-27 14:55:28 +01:00
method = self.username.split("__")[0]
return CONNECTION_METHODS.get(method, _("identifiants spécifiques"))
class Meta:
ordering = ["username"]