use forms

bda/forms.py

@@ -4,10 +4,12 @@ from __future__ import division
 from __future__ import print_function
 from __future__ import unicode_literals
 
+from datetime import timedelta
+
 from django import forms
 from django.forms.models import BaseInlineFormSet
 from django.utils import timezone
-from bda.models import Attribution
+from bda.models import Attribution, Spectacle
 
 
 class BaseBdaFormSet(BaseInlineFormSet):
@@ -36,19 +38,45 @@ class TokenForm(forms.Form):
     token = forms.CharField(widget=forms.widgets.Textarea())
 
 
-class SpectacleModelChoiceField(forms.ModelChoiceField):
+class AttributionModelMultipleChoiceField(forms.ModelMultipleChoiceField):
     def label_from_instance(self, obj):
-        return "%s le %s (%s) à %.02f€" % (obj.title, obj.date_no_seconds(),
-                                           obj.location, obj.price)
+        return "%s" % obj.spectacle
 
 
 class ResellForm(forms.Form):
-    attributions = forms.ModelMultipleChoiceField(
+    attributions = AttributionModelMultipleChoiceField(
                          queryset=Attribution.objects.none(),
-                         widget=forms.CheckboxSelectMultiple)
+                         widget=forms.CheckboxSelectMultiple,
+                         required=False)
 
     def __init__(self, participant, *args, **kwargs):
         super(ResellForm, self).__init__(*args, **kwargs)
         self.fields['attributions'].queryset = participant.attribution_set\
             .filter(spectacle__date__gte=timezone.now(),
                     revente__isnull=True)
+
+
+class AnnulForm(forms.Form):
+    attributions = AttributionModelMultipleChoiceField(
+                         queryset=Attribution.objects.none(),
+                         widget=forms.CheckboxSelectMultiple,
+                         required=False)
+
+    def __init__(self, participant, *args, **kwargs):
+        super(AnnulForm, self).__init__(*args, **kwargs)
+        self.fields['attributions'].queryset = participant.attribution_set\
+            .filter(spectacle__date__gte=timezone.now(),
+                    revente__isnull=False,
+                    revente__date__gte=timezone.now()-timedelta(hours=1))
+
+
+class InscriptionReventeForm(forms.Form):
+    spectacles = forms.ModelMultipleChoiceField(
+                    queryset=Spectacle.objects.none(),
+                    widget=forms.CheckboxSelectMultiple,
+                    required=False)
+
+    def __init__(self, tirage, *args, **kwargs):
+        super(InscriptionReventeForm, self).__init__(*args, **kwargs)
+        self.fields['spectacles'].queryset = tirage.spectacle_set.filter(
+                date__gte=timezone.now())