from django import forms

from sage_days.sdi.models import Registrant, TOPICS_CHOICES

from captcha.fields import CaptchaField

class RegisterForm(forms.ModelForm):

    """ The form that is displayed to user.

    """

    topics_interested = forms.MultipleChoiceField(widget=forms.CheckboxSelectMultiple, choices=TOPICS_CHOICES, required=False)

    class Meta:

        model = Registrant

    def clean_email(self):

        """ See if the user has already registered using the email.

        """

        email = self.cleaned_data['email'].strip()

        try:

            Registrant.objects.get(email__iexact=email)

            raise forms.ValidationError("This email is already registered. Did you register earlier??")

        except Registrant.DoesNotExist:

            return email

    def clean_topics_interested(self):

        """ Join the choices using PIPE character and store them.

        """

        topics = self.cleaned_data['topics_interested']

        return "|".join(topics)

class SearchForm(forms.Form):

    """ Search form for filtering registrants.

    """

    topics_interested = forms.CharField(required=False, help_text=" Use numbers seperated by spaces and prefix a minus to exclude")

    knowledge_of_python = forms.CharField(required=False, help_text=" Give single number from 1 to 5 or a range seperated by hyphen")

    knowledge_of_sage = forms.CharField(required=False, help_text=" Give single number from 1 to 5 or a range seperated by hyphen")

    likeliness_of_attending = forms.CharField(required=False, help_text=" Give single number from 1 to 5 or a range seperated by hyphen")

    need_for_python_workshop = forms.BooleanField(required=False)

    def clean_topics_interested(self):

        """ split and return include list and exclude list

        """

        include_list = []

        exclude_list = []

        topics = self.cleaned_data['topics_interested']

        for number in topics.split():

            if number.startswith('-'):

                exclude_list.append(number[1:])

            else:

                include_list.append(number)

        return (include_list, exclude_list)

    def clean_knowledge_of_python(self):

        """ if two numbers are given take start and stop.

        else take highest as default.

        """

        range_str = self.cleaned_data['knowledge_of_python'].strip()

        if not range_str:

            return ("","")

        start_stop = range_str.split("-")

        if len(start_stop) == 1:

            if start_stop[0].isdigit():

                return (start_stop[0], "")

            else:

                raise forms.ValidationError("Invalid range for knowledge of Python")

        elif len(start_stop) == 2:

            start, stop = start_stop

            if start.isdigit() and stop.isdigit():

                return (start, stop)

            else:

                 raise forms.ValidationError("Invalid range for knowledge of Python")

        else:

            raise forms.ValidationError("Invalid range for knowledge of Python")

    def clean_knowledge_of_sage(self):

        """ if two numbers are given take start and stop.

        else take highest as default.

        """

        range_str = self.cleaned_data['knowledge_of_sage'].strip()

        if not range_str:

            return ("","")

        start_stop = range_str.split("-")

        if len(start_stop) == 1:

            if start_stop[0].isdigit():

                return (start_stop[0], "")

            else:

                raise forms.ValidationError("Invalid range for knowledge of Sage")

        elif len(start_stop) == 2:

            start, stop = start_stop

            if start.isdigit() and stop.isdigit():

                return (start, stop)

            else:

                 raise forms.ValidationError("Invalid range for knowledge of Sage")

        else:

            raise forms.ValidationError("Invalid range for knowledge of Sage")

    def clean_likeliness_of_attending(self):

        """ if two numbers are given take start and stop.

        else take highest as default.

        """

        range_str = self.cleaned_data['likeliness_of_attending'].strip()

        if not range_str:

            return ("","")

        start_stop = range_str.split("-")

        if len(start_stop) == 1:

            if start_stop[0].isdigit():

                return (start_stop[0], "")

            else:

                raise forms.ValidationError("Invalid range for Likeliness of attending the workshop")

        elif len(start_stop) == 2:

            start, stop = start_stop

            if start.isdigit() and stop.isdigit():

                return (start, stop)

            else:

                 raise forms.ValidationError("Invalid range for Likeliness of attending the workshop")

        else:

            raise forms.ValidationError("Invalid range for Likeliness of attending the workshop")