# -*- coding: utf-8 -*-

"""

BR-specific Form helpers

"""

from django.forms import ValidationError

from django.forms.fields import Field, RegexField, CharField, Select, EMPTY_VALUES

from django.utils.encoding import smart_unicode

from django.utils.translation import ugettext_lazy as _

import re

try:

    set

except NameError:

    from sets import Set as set     # For Python 2.3

phone_digits_re = re.compile(r'^(\d{2})[-\.]?(\d{4})[-\.]?(\d{4})$')

class BRZipCodeField(RegexField):

    default_error_messages = {

        'invalid': _('Enter a zip code in the format XXXXX-XXX.'),

    }

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

        super(BRZipCodeField, self).__init__(r'^\d{5}-\d{3}$',

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

class BRPhoneNumberField(Field):

    default_error_messages = {

        'invalid': _('Phone numbers must be in XX-XXXX-XXXX format.'),

    }

    def clean(self, value):

        super(BRPhoneNumberField, self).clean(value)

        if value in EMPTY_VALUES:

            return u''

        value = re.sub('(\(|\)|\s+)', '', smart_unicode(value))

        m = phone_digits_re.search(value)

        if m:

            return u'%s-%s-%s' % (m.group(1), m.group(2), m.group(3))

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

class BRStateSelect(Select):

    """

    A Select widget that uses a list of Brazilian states/territories

    as its choices.

    """

    def __init__(self, attrs=None):

        from br_states import STATE_CHOICES

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

class BRStateChoiceField(Field):

    """

    A choice field that uses a list of Brazilian states as its choices.

    """

    widget = Select

    default_error_messages = {

        'invalid': _(u'Select a valid brazilian state. That state is not one of the available states.'),

    }

    def __init__(self, required=True, widget=None, label=None,

                 initial=None, help_text=None):

        super(BRStateChoiceField, self).__init__(required, widget, label,

                                                 initial, help_text)

        from br_states import STATE_CHOICES

        self.widget.choices = STATE_CHOICES

    def clean(self, value):

        value = super(BRStateChoiceField, self).clean(value)

        if value in EMPTY_VALUES:

            value = u''

        value = smart_unicode(value)

        if value == u'':

            return value

        valid_values = set([smart_unicode(k) for k, v in self.widget.choices])

        if value not in valid_values:

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

        return value

def DV_maker(v):

    if v >= 2:

        return 11 - v

    return 0

class BRCPFField(CharField):

    """

    This field validate a CPF number or a CPF string. A CPF number is

    compounded by XXX.XXX.XXX-VD. The two last digits are check digits.

    More information:

    http://en.wikipedia.org/wiki/Cadastro_de_Pessoas_F%C3%ADsicas

    """

    default_error_messages = {

        'invalid': _("Invalid CPF number."),

        'max_digits': _("This field requires at most 11 digits or 14 characters."),

        'digits_only': _("This field requires only numbers."),

    }

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

        super(BRCPFField, self).__init__(max_length=14, min_length=11, *args, **kwargs)

    def clean(self, value):

        """

        Value can be either a string in the format XXX.XXX.XXX-XX or an

        11-digit number.

        """

        value = super(BRCPFField, self).clean(value)

        if value in EMPTY_VALUES:

            return u''

        orig_value = value[:]

        if not value.isdigit():

            value = re.sub("[-\.]", "", value)

        try:

            int(value)

        except ValueError:

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

        if len(value) != 11:

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

        orig_dv = value[-2:]

        new_1dv = sum([i * int(value[idx]) for idx, i in enumerate(range(10, 1, -1))])

        new_1dv = DV_maker(new_1dv % 11)

        value = value[:-2] + str(new_1dv) + value[-1]

        new_2dv = sum([i * int(value[idx]) for idx, i in enumerate(range(11, 1, -1))])

        new_2dv = DV_maker(new_2dv % 11)

        value = value[:-1] + str(new_2dv)

        if value[-2:] != orig_dv:

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

        return orig_value

class BRCNPJField(Field):

    default_error_messages = {

        'invalid': _("Invalid CNPJ number."),

        'digits_only': _("This field requires only numbers."),

        'max_digits': _("This field requires at least 14 digits"),

    }

    def clean(self, value):

        """

        Value can be either a string in the format XX.XXX.XXX/XXXX-XX or a

        group of 14 characters.

        """

        value = super(BRCNPJField, self).clean(value)

        if value in EMPTY_VALUES:

            return u''

        orig_value = value[:]

        if not value.isdigit():

            value = re.sub("[-/\.]", "", value)

        try:

            int(value)

        except ValueError:

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

        if len(value) != 14:

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

        orig_dv = value[-2:]

        new_1dv = sum([i * int(value[idx]) for idx, i in enumerate(range(5, 1, -1) + range(9, 1, -1))])

        new_1dv = DV_maker(new_1dv % 11)

        value = value[:-2] + str(new_1dv) + value[-1]

        new_2dv = sum([i * int(value[idx]) for idx, i in enumerate(range(6, 1, -1) + range(9, 1, -1))])

        new_2dv = DV_maker(new_2dv % 11)

        value = value[:-1] + str(new_2dv)

        if value[-2:] != orig_dv:

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

        return orig_value