kadenios/elections/forms.py

139 lines
4.1 KiB
Python
Raw Normal View History

2021-04-15 17:26:27 +02:00
from translated_fields import language_code_formfield_callback
2020-12-18 00:19:18 +01:00
from django import forms
from django.forms.models import inlineformset_factory
2020-12-19 22:22:23 +01:00
from django.utils import timezone
from django.utils.translation import gettext_lazy as _
2020-12-18 00:19:18 +01:00
2020-12-19 22:22:23 +01:00
from .models import Election, Option, Question
from .utils import check_csv
2020-12-19 22:22:23 +01:00
class ElectionForm(forms.ModelForm):
formfield_callback = language_code_formfield_callback
2020-12-19 22:22:23 +01:00
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 = [
*Election.name.fields,
"start_date",
"end_date",
*Election.description.fields,
"restricted",
*Election.vote_restrictions.fields,
]
widgets = {
"description_en": forms.Textarea(attrs={"rows": 4}),
"description_fr": forms.Textarea(attrs={"rows": 4}),
"vote_restrictions_en": forms.Textarea(attrs={"rows": 4}),
"vote_restrictions_fr": forms.Textarea(attrs={"rows": 4}),
}
2020-12-18 00:19:18 +01:00
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."),
)
2020-12-23 17:08:07 +01:00
csv_file.seek(0)
return csv_file
class VoterMailForm(forms.Form):
objet = forms.CharField()
message = forms.CharField(widget=forms.Textarea)
class QuestionForm(forms.ModelForm):
formfield_callback = language_code_formfield_callback
class Meta:
model = Question
fields = [*Question.text.fields, "type"]
widgets = {"text_fr": forms.TextInput, "text_en": forms.TextInput}
class OptionForm(forms.ModelForm):
formfield_callback = language_code_formfield_callback
class Meta:
model = Option
fields = [*Option.text.fields]
widgets = {"text_fr": forms.TextInput, "text_en": forms.TextInput}
class DeleteVoteForm(forms.Form):
def __init__(self, **kwargs):
voter = kwargs.pop("voter", None)
super().__init__(**kwargs)
if voter is not None:
self.fields["delete"].label = _("Supprimer le vote de {} ({}) ?").format(
voter.full_name, voter.get_username()
)
delete = forms.ChoiceField(
label=_("Supprimer"), choices=(("non", _("Non")), ("oui", _("Oui")))
)
class SelectVoteForm(forms.ModelForm):
2020-12-18 00:19:18 +01:00
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)
class Meta:
model = Option
fields = []
exclude = ["voters"]
2020-12-18 00:19:18 +01:00
class RankVoteForm(forms.ModelForm):
def __init__(self, **kwargs):
super().__init__(**kwargs)
# We set the option's text as the label for the rank
instance = kwargs.get("instance", None)
if instance is not None:
self.fields["rank"].label = instance.text
rank = forms.IntegerField(required=False)
class Meta:
model = Option
fields = []
exclude = ["voters"]
class BallotFormset:
select = inlineformset_factory(
Question, Option, extra=0, form=SelectVoteForm, can_delete=False
)
rank = inlineformset_factory(
Question, Option, extra=0, form=RankVoteForm, can_delete=False
)