kpsul/bda/forms.py

# coding: utf-8

from django import forms
from django.forms.models import BaseInlineFormSet
from bda.models import Spectacle, Participant, ChoixSpectacle, Attribution

class BaseBdaFormSet(BaseInlineFormSet):
    def clean(self):
        """Checks that no two articles have the same title."""
        super(BaseBdaFormSet, self).clean()
        if any(self.errors):
            # Don't bother validating the formset unless each form is valid on its own
            return
        spectacles = []
        for i in range(0, self.total_form_count()):
            form = self.forms[i]
            if not form.cleaned_data:
                continue
            spectacle = form.cleaned_data['spectacle']
            delete = form.cleaned_data['DELETE']
            if not delete and spectacle in spectacles:
                raise forms.ValidationError("Vous ne pouvez pas vous " + \
                        "inscrire deux fois pour le même spectacle.")
            spectacles.append(spectacle)

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

class SpectacleModelChoiceField(forms.ModelChoiceField):
    def label_from_instance(self, obj):
        return u"%s le %s (%s) à %.02f€" % (obj.title, obj.date_no_seconds(),
                                            obj.location, obj.price)

class ResellForm(forms.Form):
    count = forms.ChoiceField(choices = (("1","1"),("2","2"),))
    spectacle = SpectacleModelChoiceField(queryset=Spectacle.objects.none())

    def __init__(self, participant, *args, **kwargs):
        super(ResellForm, self).__init__(*args, **kwargs)
        self.fields['spectacle'].queryset = participant.attributions.all().distinct()