Ticket #3908: brcpffield_add.diff

contrib/localflavor/br/forms.py

@@ -5 +5 @@
 
 from django.newforms import ValidationError
 from django.newforms.fields import Field, RegexField, Select, EMPTY_VALUES
+from django.newforms.fields import CharField
 from django.newforms.util import smart_unicode
 from django.utils.translation import gettext
 import re
@@ -27 +28 @@
         m = phone_digits_re.search(value)
         if m:
             return u'%s-%s-%s' % (m.group(1), m.group(2), m.group(3))
-        raise ValidationError(gettext(u'Phone numbers must be in XX-XXXX-XXXX format.'))
+        raise ValidationError(gettext(u'Phone numbers must be in XX-XXXX-XXXX 
+format.'))
 
 class BRStateSelect(Select):
     """
@@ -37 +39 @@
     def __init__(self, attrs=None):
         from br_states import STATE_CHOICES # relative import
         super(BRStateSelect, self).__init__(attrs, choices=STATE_CHOICES)
+        
+class BRCPFField(CharField):
+    def __init__(self, *args, **kwargs):
+        super(BRCPFField, self).__init__(max_length=11, *args, **kwargs)
+
+    def clean(self, value):
+        super(BRCPFField, self).clean(value)
+        if value in EMPTY_VALUES:
+            return u''
+        try:
+            int(value)
+        except ValueError:
+             raise ValidationError(gettext(u'The CPF field must contain only 
+numbers.'))
+        if self._validate_cpf(value):
+            return value
+
+    def _validate_cpf(self, cpf):
+        if len(cpf) < 11:
+            raise ValidationError(gettext(u'Invalid CPF. The CPF field must 
+contain 11 digits.'))
+        number = 0
+        for i,j in enumerate(range(10,1,-1)):
+            number += int(cpf[i])*j
+        if self._create_dv(number) <> cpf[9]:
+            raise ValidationError(gettext(u'Invalid CPF.'))
+
+        number = 0
+        
+        for i,j in enumerate(range(11,1,-1)):
+            number += int(cpf[i])*j
+        if self._create_dv(number) <> cpf[10]:
+            raise ValidationError(gettext(u'Invalid CPF.'))
+        return True
+
+    def _create_dv(self, number):
+        if number % 11 > 1:
+            return str(11 - number % 11)
+        else:
+            return '0'