from django import forms
from openair.estimate.models import MyModel

MAX_LEG_FORMS = 5

class MyModelForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(MyModelForm, self).__init__(*args, **kwargs)
        for i in range(0, MAX_LEG_FORMS):
            self.fields['leg_location_%s' % i] = forms.CharField(
                label=u'Location',
                max_length=50,
                required=False
            )
            self.fields['leg_airport_%s' % i] = forms.RegexField(
                label=u'Departure Airport',
                max_length=3,
                min_length=3,
                regex=r'^[a-zA-Z0-9]{3}$',
                required=False
            )
            self.fields['leg_delay_%s' % i] = forms.IntegerField(
                label=u'Delay (in hours)',
                min_value=0,
                initial=0,
                required=False
            )

    def clean(self):
        cleaned_data = self.cleaned_data
        for i in range(1, MAX_LEG_FORMS):
            location = cleaned_data.get("leg_location_%s" % i)
            airport = cleaned_data.get("leg_airport_%s" % i)
            delay = cleaned_data.get("leg_delay_%s" % i)
            if location != '' and airport in ['', False, None]:
                airport_msg = 'Validation message here'
                self._errors["leg_airport_%s" % i] = self.error_class([airport_msg])
            if location != '' and delay in ['', False, None]:
                delay_msg = 'Validation message here'
                self._errors["leg_delay_%s" % i] = self.error_class([delay_msg])
        return cleaned_data

    class Meta:
        model = MyModel