81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
from django import forms
|
|
from django.forms.models import inlineformset_factory
|
|
from django.utils import timezone
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from .models import Election, Option, Question
|
|
from .utils import check_csv
|
|
|
|
|
|
class ElectionForm(forms.ModelForm):
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
if cleaned_data["start_date"] < timezone.now():
|
|
self.add_error(
|
|
"start_date", _("Impossible de faire débuter l'élection dans le passé")
|
|
)
|
|
elif cleaned_data["start_date"] >= cleaned_data["end_date"]:
|
|
self.add_error(
|
|
"end_date", _("Impossible de terminer l'élection avant de la commencer")
|
|
)
|
|
return cleaned_data
|
|
|
|
class Meta:
|
|
model = Election
|
|
fields = ["name", "description", "restricted", "start_date", "end_date"]
|
|
|
|
|
|
class UploadVotersForm(forms.Form):
|
|
csv_file = forms.FileField(label=_("Sélectionnez un fichier .csv"))
|
|
|
|
def clean_csv_file(self):
|
|
csv_file = self.cleaned_data["csv_file"]
|
|
if csv_file.name.lower().endswith(".csv"):
|
|
for error in check_csv(csv_file):
|
|
self.add_error(None, error)
|
|
else:
|
|
self.add_error(
|
|
None,
|
|
_("Extension de fichier invalide, il faut un fichier au format CSV."),
|
|
)
|
|
csv_file.seek(0)
|
|
return csv_file
|
|
|
|
|
|
class QuestionForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Question
|
|
fields = ["text"]
|
|
|
|
|
|
class OptionForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Option
|
|
fields = ["text"]
|
|
|
|
|
|
class VoteForm(forms.ModelForm):
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
# We set the option's text as the label for the checkbox
|
|
instance = kwargs.get("instance", None)
|
|
if instance is not None:
|
|
self.fields["selected"].label = instance.text
|
|
|
|
selected = forms.BooleanField(required=False)
|
|
|
|
def record_vote(self, user):
|
|
self.full_clean()
|
|
if self.cleaned_data["selected"]:
|
|
self.instance.voters.add(user)
|
|
else:
|
|
self.instance.voters.remove(user)
|
|
|
|
class Meta:
|
|
model = Option
|
|
fields = []
|
|
|
|
|
|
OptionFormSet = inlineformset_factory(
|
|
Question, Option, extra=0, form=VoteForm, can_delete=False
|
|
)
|