106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
AR-specific Form helpers.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from django.newforms import ValidationError
|
|||
|
|
from django.newforms.fields import RegexField, CharField, Select, EMPTY_VALUES
|
|||
|
|
from django.utils.encoding import smart_unicode
|
|||
|
|
from django.utils.translation import ugettext
|
|||
|
|
import re
|
|||
|
|
|
|||
|
|
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<EFBFBD> NNNN Postal Code or a CPA.
|
|||
|
|
|
|||
|
|
See http://www.correoargentino.com.ar/consulta_cpa/home.php
|
|||
|
|
"""
|
|||
|
|
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,
|
|||
|
|
error_message=ugettext("Enter a postal code in the format NNNN or ANNNNAAA."),
|
|||
|
|
*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(ugettext("Enter a postal code in the format NNNN or ANNNNAAA."))
|
|||
|
|
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<EFBFBD> (DNI) numbers.
|
|||
|
|
"""
|
|||
|
|
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(ugettext("This field requires only numbers."))
|
|||
|
|
if len(value) not in (7, 8):
|
|||
|
|
raise ValidationError(
|
|||
|
|
ugettext("This field requires 7 or 8 digits."))
|
|||
|
|
|
|||
|
|
return value
|
|||
|
|
|
|||
|
|
class ARCUITField(RegexField):
|
|||
|
|
"""
|
|||
|
|
This field validates a CUIT (C<EFBFBD>digo <EFBFBD>nico de Identificaci<EFBFBD>n Tributaria). A
|
|||
|
|
CUIT is of the form XX-XXXXXXXX-V. The last digit is a check digit.
|
|||
|
|
"""
|
|||
|
|
def __init__(self, *args, **kwargs):
|
|||
|
|
super(ARCUITField, self).__init__(r'^\d{2}-?\d{8}-?\d$',
|
|||
|
|
error_message=ugettext('Enter a valid CUIT in XX-XXXXXXXX-X or XXXXXXXXXXXX format.'),
|
|||
|
|
*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(ugettext("Invalid CUIT."))
|
|||
|
|
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)
|
|||
|
|
|