# HG changeset patch
# User Madhusudan.C.S <madhusudancs@gmail.com>
# Date 1289157508 -19800
# Node ID 97cbeda302a90bf811b09cfbff0966cfe068be14
# Parent  92fbbbc4c97a1256e7751acfb0090b23efb9d989
Added accommodation Django form, wrote the save method and overwrote the cleaner.

diff -r 92fbbbc4c97a -r 97cbeda302a9 project/scipycon/registration/forms.py
--- a/project/scipycon/registration/forms.py	Mon Nov 08 00:48:45 2010 +0530
+++ b/project/scipycon/registration/forms.py	Mon Nov 08 00:48:28 2010 +0530
@@ -3,6 +3,7 @@
 
 from project.scipycon.registration.models import SIZE_CHOICES
 from project.scipycon.registration.models import OCCUPATION_CHOICES
+from project.scipycon.registration.models import Accommodation
 from project.scipycon.registration.models import Wifi
 
 
@@ -78,6 +79,54 @@
         model = Wifi
         fields = ('wifi',)
 
+
+class AccommodationForm(forms.ModelForm):
+    """PyCon Accommodation form
+    """
+
+    def save(self, user, scope):
+        try:
+            acco = Accommodation.objects.get(user=user, scope=scope)
+        except ObjectDoesNotExist:
+            acco = Accommodation(user=user, scope=scope)
+
+        sex = self.cleaned_data['sex']
+        accommodation_required = self.cleaned_data['accommodation_required']
+        accommodation_days = self.cleaned_data['accommodation_days']
+
+        acco.sex = sex
+        acco.accommodation_required = accommodation_required
+        acco.accommodation_days = accommodation_days if (
+            accommodation_days) else 0
+
+        acco.save()
+
+        return acco
+
+    def clean(self):
+        """Makes sure that accommodation form is correct, i.e. sex
+        and number of days required are filled in when the accommodation
+        is required.
+        """
+        cleaned = self.cleaned_data
+
+        sex = self.cleaned_data['sex']
+        accommodation_required = self.cleaned_data['accommodation_required']
+        accommodation_days = self.cleaned_data['accommodation_days']
+
+        if accommodation_required and (not sex or not accommodation_days
+            or accommodation_days == 0):
+            raise forms.ValidationError(
+                u"If accommodation is required both gender and number of "
+                "days for which accommodation is required is mandatory.")
+
+        return super(AccommodationForm, self).clean()
+
+    class Meta:
+        model = Accommodation
+        fields = ('accommodation_required', 'sex', 'accommodation_days')
+
+
 PC = (
         ('all', 'all'),
         ('paid', 'paid'),