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

"""

AR-specific Form helpers.

"""

from django.forms import ValidationError

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

from django.utils.encoding import smart_unicode

from django.utils.translation import ugettext_lazy as _

class ARProvinceSelect(Select):

    """

    A Select widget that uses a list of Argentinean provinces/autonomous cities

    as its choices.

    """

    def __init__(self, attrs=None):

        from ar_provinces import PROVINCE_CHOICES

        super(ARProvinceSelect, self).__init__(attrs, choices=PROVINCE_CHOICES)

class ARPostalCodeField(RegexField):

    """

    A field that accepts a 'classic' NNNN Postal Code or a CPA.

    See http://www.correoargentino.com.ar/consulta_cpa/home.php

    """

    default_error_messages = {

        'invalid': _("Enter a postal code in the format NNNN or ANNNNAAA."),

    }

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

        super(ARPostalCodeField, self).__init__(r'^\d{4}$|^[A-HJ-NP-Za-hj-np-z]\d{4}\D{3}$',

            min_length=4, max_length=8, *args, **kwargs)

    def clean(self, value):

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

        if value in EMPTY_VALUES:

            return u''

        if len(value) not in (4, 8):

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

        if len(value) == 8:

            return u'%s%s%s' % (value[0].upper(), value[1:5], value[5:].upper())

        return value

class ARDNIField(CharField):

    """

    A field that validates 'Documento Nacional de Identidad' (DNI) numbers.

    """

    default_error_messages = {

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

        'max_digits': _("This field requires 7 or 8 digits."),

    }

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

        super(ARDNIField, self).__init__(max_length=10, min_length=7, *args,

                **kwargs)

    def clean(self, value):

        """

        Value can be a string either in the [X]X.XXX.XXX or [X]XXXXXXX formats.

        """

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

        if value in EMPTY_VALUES:

            return u''

        if not value.isdigit():

            value = value.replace('.', '')

        if not value.isdigit():

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

        if len(value) not in (7, 8):

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

        return value

class ARCUITField(RegexField):

    """

    This field validates a CUIT (Código Único de Identificación Tributaria). A

    CUIT is of the form XX-XXXXXXXX-V. The last digit is a check digit.

    """

    default_error_messages = {

        'invalid': _('Enter a valid CUIT in XX-XXXXXXXX-X or XXXXXXXXXXXX format.'),

        'checksum': _("Invalid CUIT."),

    }

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

        super(ARCUITField, self).__init__(r'^\d{2}-?\d{8}-?\d$',

            *args, **kwargs)

    def clean(self, value):

        """

        Value can be either a string in the format XX-XXXXXXXX-X or an

        11-digit number.

        """

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

        if value in EMPTY_VALUES:

            return u''

        value, cd = self._canon(value)

        if self._calc_cd(value) != cd:

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

        return self._format(value, cd)

    def _canon(self, cuit):

        cuit = cuit.replace('-', '')

        return cuit[:-1], cuit[-1]

    def _calc_cd(self, cuit):

        mults = (5, 4, 3, 2, 7, 6, 5, 4, 3, 2)

        tmp = sum([m * int(cuit[idx]) for idx, m in enumerate(mults)])

        return str(11 - tmp % 11)

    def _format(self, cuit, check_digit=None):

        if check_digit == None:

            check_digit = cuit[-1]

            cuit = cuit[:-1]

        return u'%s-%s-%s' % (cuit[:2], cuit[2:], check_digit)