kpsul/bda/forms.py

from django import forms
from django.forms.models import BaseInlineFormSet
from django.utils import timezone

from bda.models import Attribution, Spectacle, SpectacleRevente


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')
        choices = [(sp.pk, str(sp)) for sp in spectacles]
        self.force_choices('spectacle', choices)

    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
            else:
                field.choices = choices


class TokenForm(forms.Form):
    token = forms.CharField(widget=forms.widgets.Textarea())


class AttributionModelMultipleChoiceField(forms.ModelMultipleChoiceField):
    def label_from_instance(self, obj):
        return str(obj.spectacle)


class ReventeModelMultipleChoiceField(forms.ModelMultipleChoiceField):
    def __init__(self, *args, own=True, **kwargs):
        super().__init__(*args, **kwargs)
        self.own = own

    def label_from_instance(self, obj):
        label = "{show}{suffix}"
        suffix = ""
        if self.own:
            # C'est notre propre revente : pas besoin de spécifier le vendeur
            if obj.soldTo is not None:
                suffix = " -- Vendue à {firstname} {lastname}".format(
                            firstname=obj.soldTo.user.first_name,
                            lastname=obj.soldTo.user.last_name,
                        )
        else:
            # Ce n'est pas à nous : on ne voit jamais l'acheteur
            suffix = " -- Vendue par {firstname} {lastname}".format(
                            firstname=obj.seller.user.first_name,
                            lastname=obj.seller.user.last_name,
                        )

        return label.format(show=str(obj.attribution.spectacle),
                            suffix=suffix)


class ResellForm(forms.Form):
    attributions = AttributionModelMultipleChoiceField(
                         label='',
                         queryset=Attribution.objects.none(),
                         widget=forms.CheckboxSelectMultiple,
                         required=False)

    def __init__(self, participant, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['attributions'].queryset = (
            participant.attribution_set
            .filter(spectacle__date__gte=timezone.now())
            .exclude(revente__seller=participant)
            .select_related('spectacle', 'spectacle__location',
                            'participant__user')
        )


class AnnulForm(forms.Form):
    reventes = ReventeModelMultipleChoiceField(
                         own=True,
                         label='',
                         queryset=Attribution.objects.none(),
                         widget=forms.CheckboxSelectMultiple,
                         required=False)

    def __init__(self, participant, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['reventes'].queryset = (
            participant.original_shows
            .filter(attribution__spectacle__date__gte=timezone.now(),
                    notif_sent=False,
                    soldTo__isnull=True)
            .select_related('attribution__spectacle',
                            'attribution__spectacle__location')
        )


class InscriptionReventeForm(forms.Form):
    spectacles = forms.ModelMultipleChoiceField(
                    queryset=Spectacle.objects.none(),
                    widget=forms.CheckboxSelectMultiple,
                    required=False)

    def __init__(self, tirage, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['spectacles'].queryset = (
            tirage.spectacle_set
            .select_related('location')
            .filter(date__gte=timezone.now())
        )


class ReventeTirageAnnulForm(forms.Form):
    reventes = ReventeModelMultipleChoiceField(
                  own=False,
                  label='',
                  queryset=SpectacleRevente.objects.none(),
                  widget=forms.CheckboxSelectMultiple,
                  required=False
              )

    def __init__(self, participant, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['reventes'].queryset = (
            participant.entered.filter(soldTo__isnull=True)
                       .select_related('attribution__spectacle',
                                       'seller__user')
        )


class ReventeTirageForm(forms.Form):
    reventes = ReventeModelMultipleChoiceField(
                  own=False,
                  label='',
                  queryset=SpectacleRevente.objects.none(),
                  widget=forms.CheckboxSelectMultiple,
                  required=False
              )

    def __init__(self, participant, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['reventes'].queryset = (
            SpectacleRevente.objects.filter(
                notif_sent=True,
                shotgun=False,
                tirage_done=False
            ).exclude(confirmed_entry=participant)
            .select_related('attribution__spectacle')
        )


class SoldForm(forms.Form):
    reventes = ReventeModelMultipleChoiceField(
                        own=True,
                        label='',
                        queryset=Attribution.objects.none(),
                        widget=forms.CheckboxSelectMultiple)

    def __init__(self, participant, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['reventes'].queryset = (
            participant.original_shows
            .filter(soldTo__isnull=False)
            .exclude(soldTo=participant)
            .select_related('attribution__spectacle',
                            'attribution__spectacle__location')
        )