gestioCOF/bda/forms.py

185 lines
6.5 KiB
Python
Raw Normal View History

from django import forms
2017-04-21 18:22:53 +02:00
from django.forms.models import BaseInlineFormSet
2018-11-21 16:14:52 +01:00
from django.template import loader
2016-07-25 23:03:33 +02:00
from django.utils import timezone
2017-04-21 18:22:53 +02:00
2020-10-29 11:14:06 +01:00
from bda.models import SpectacleRevente
2017-04-21 18:22:53 +02:00
class InscriptionInlineFormSet(BaseInlineFormSet):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
# self.instance is a Participant object
tirage = self.instance.tirage
# set once for all "spectacle" field choices
# - restrict choices to the spectacles of this tirage
# - force_choices avoid many db requests
spectacles = tirage.spectacle_set.select_related("location")
2017-04-21 18:22:53 +02:00
choices = [(sp.pk, str(sp)) for sp in spectacles]
self.force_choices("spectacle", choices)
2017-04-21 18:22:53 +02:00
def force_choices(self, name, choices):
"""Set choices of a field.
As ModelChoiceIterator (default use to get choices of a
ModelChoiceField), it appends an empty selection if requested.
"""
for form in self.forms:
field = form.fields[name]
if field.empty_label is not None:
field.choices = [("", field.empty_label)] + choices
2017-04-21 18:22:53 +02:00
else:
field.choices = choices
class TokenForm(forms.Form):
token = forms.CharField(widget=forms.widgets.Textarea())
2019-01-07 15:27:41 +01:00
class TemplateLabelField(forms.ModelMultipleChoiceField):
"""
Extends ModelMultipleChoiceField to offer two more customization options :
- `label_from_instance` can be used with a template file
- the widget rendering template can be specified with `option_template_name`
"""
def __init__(
self,
label_template_name=None,
context_object_name="obj",
option_template_name=None,
*args,
**kwargs
):
super().__init__(*args, **kwargs)
self.label_template_name = label_template_name
self.context_object_name = context_object_name
if option_template_name is not None:
self.widget.option_template_name = option_template_name
def label_from_instance(self, obj):
if self.label_template_name is None:
return super().label_from_instance(obj)
else:
return loader.render_to_string(
self.label_template_name, context={self.context_object_name: obj}
)
# Formulaires pour revente_manage
2017-10-26 12:40:11 +02:00
class ResellForm(forms.Form):
def __init__(self, participant, *args, **kwargs):
2018-01-16 16:22:52 +01:00
super().__init__(*args, **kwargs)
2019-01-07 16:24:30 +01:00
self.fields["attributions"] = TemplateLabelField(
queryset=participant.attribution_set.filter(
spectacle__date__gte=timezone.now(), paid=True
2019-01-07 16:24:30 +01:00
)
.exclude(revente__seller=participant)
2019-01-07 16:24:30 +01:00
.select_related("spectacle", "spectacle__location", "participant__user"),
widget=forms.CheckboxSelectMultiple,
required=False,
label_template_name="bda/forms/attribution_label_table.html",
option_template_name="bda/forms/checkbox_table.html",
context_object_name="attribution",
2017-04-08 12:10:23 +02:00
)
2016-07-27 13:08:00 +02:00
2017-10-26 12:40:11 +02:00
2016-07-27 13:08:00 +02:00
class AnnulForm(forms.Form):
def __init__(self, participant, *args, **kwargs):
2018-01-16 16:22:52 +01:00
super().__init__(*args, **kwargs)
2019-01-07 16:24:30 +01:00
self.fields["reventes"] = TemplateLabelField(
label="",
queryset=participant.original_shows.filter(
attribution__spectacle__date__gte=timezone.now(), soldTo__isnull=True
)
.select_related(
"attribution__spectacle", "attribution__spectacle__location"
)
.order_by("-date"),
widget=forms.CheckboxSelectMultiple,
required=False,
2019-01-07 16:24:30 +01:00
label_template_name="bda/forms/revente_self_label_table.html",
option_template_name="bda/forms/checkbox_table.html",
context_object_name="revente",
2017-04-08 12:10:23 +02:00
)
2016-07-27 13:08:00 +02:00
2017-10-26 12:40:11 +02:00
class SoldForm(forms.Form):
def __init__(self, participant, *args, **kwargs):
super().__init__(*args, **kwargs)
self.fields["reventes"] = TemplateLabelField(
queryset=participant.original_shows.filter(soldTo__isnull=False)
.exclude(soldTo=participant)
.select_related(
"attribution__spectacle", "attribution__spectacle__location"
),
widget=forms.CheckboxSelectMultiple,
label_template_name="bda/forms/revente_sold_label_table.html",
option_template_name="bda/forms/checkbox_table.html",
context_object_name="revente",
)
# Formulaire pour revente_subscribe
2016-07-27 13:08:00 +02:00
class InscriptionReventeForm(forms.Form):
def __init__(self, tirage, *args, **kwargs):
2018-01-16 16:22:52 +01:00
super().__init__(*args, **kwargs)
self.fields["spectacles"] = TemplateLabelField(
queryset=tirage.spectacle_set.select_related("location").filter(
date__gte=timezone.now()
),
widget=forms.CheckboxSelectMultiple,
required=False,
label_template_name="bda/forms/spectacle_label_table.html",
option_template_name="bda/forms/checkbox_table.html",
context_object_name="spectacle",
)
2017-02-16 04:52:44 +01:00
2017-10-26 12:40:11 +02:00
# Formulaires pour revente_tirages
class ReventeTirageAnnulForm(forms.Form):
def __init__(self, participant, *args, **kwargs):
super().__init__(*args, **kwargs)
self.fields["reventes"] = TemplateLabelField(
queryset=participant.entered.filter(soldTo__isnull=True).select_related(
"attribution__spectacle", "seller__user"
),
widget=forms.CheckboxSelectMultiple,
required=False,
label_template_name="bda/forms/revente_other_label_table.html",
option_template_name="bda/forms/checkbox_table.html",
context_object_name="revente",
)
class ReventeTirageForm(forms.Form):
def __init__(self, participant, *args, **kwargs):
super().__init__(*args, **kwargs)
2018-12-07 17:35:40 +01:00
self.fields["reventes"] = TemplateLabelField(
queryset=(
SpectacleRevente.objects.filter(
notif_sent=True,
shotgun=False,
tirage_done=False,
attribution__spectacle__tirage=participant.tirage,
2018-12-07 17:35:40 +01:00
)
.exclude(confirmed_entry=participant)
.select_related("attribution__spectacle")
),
widget=forms.CheckboxSelectMultiple,
required=False,
label_template_name="bda/forms/revente_other_label_table.html",
option_template_name="bda/forms/checkbox_table.html",
context_object_name="revente",
)