"""

India-specific Form helpers.

"""

from django.newforms import ValidationError

from django.newforms.fields import Field, RegexField, Select, EMPTY_VALUES

from django.utils.encoding import smart_unicode

from django.utils.translation import gettext

import re

class INZipCodeField(RegexField):

    default_error_messages = {

        'invalid': gettext(u'Enter a zip code in the format XXXXXXX.'),

    }

    def __init__(self, *args, **kwargs):

        super(INZipCodeField, self).__init__(r'^\d{6}$',

            max_length=None, min_length=None, *args, **kwargs)

class INStateField(Field):

    """

    A form field that validates its input is a Indian state name or

    abbreviation. It normalizes the input to the standard two-letter vehicle

    registration abbreviation for the given state or union territory

    """

    default_error_messages = {

        'invalid': u'Enter a Indian state or territory.',

    }

    def clean(self, value):

        from in_states import STATES_NORMALIZED

        super(INStateField, self).clean(value)

        if value in EMPTY_VALUES:

            return u''

        try:

            value = value.strip().lower()

        except AttributeError:

            pass

        else:

            try:

                return smart_unicode(STATES_NORMALIZED[value.strip().lower()])

            except KeyError:

                pass

        raise ValidationError(self.error_messages['invalid'])

class INStateSelect(Select):

    """

    A Select widget that uses a list of Indian states/territories as its

    choices.

    """

    def __init__(self, attrs=None):

        from in_states import STATE_CHOICES

        super(INStateSelect, self).__init__(attrs, choices=STATE_CHOICES)