| # -*- 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 |