Ticket #4122: localflavor_cl.diff

django/contrib/localflavor/cl/forms.py

@@ +1 @@
+"""
+Chile specific form helpers.
+"""
+from django.newforms import ValidationError
+from django.newforms.fields import RegexField, EMPTY_VALUES
+from django.utils.translation import gettext
+
+class CLRutField(RegexField):
+    """
+    Chilean "Rol Unico Tributario" (RUT) field. This is the chilean national identification number.
+    """
+    def __init__(self, *args, **kwargs):
+        if kwargs.get('strict')  == True:
+            del kwargs['strict']
+            super(CLRutField, self).__init__(r'^(\d{1,2}\.)?\d{3}\.\d{3}-[\dkK]$',
+                error_message=gettext(u'Enter valid a Chilean Rut. The format is XX.XXX.XXX-X.'),
+                *args, **kwargs)
+        else: # We take it relaxed and format it     for the user,
+                  # accepting RUTs that are validate, but do not exist in the real world.
+            super(CLRutField, self).__init__(r'^[\d\.]{1,11}-?[\dkK]$',
+                error_message=gettext(u'Enter valid a Chilean RUT'),
+                *args, **kwargs)
+        
+    def clean(self, value):
+        """
+        Check and clean the chilean rut.
+        """
+        super(CLRutField, self).clean(value)
+        if value in EMPTY_VALUES:
+            return u''
+        (rut, verificador) = self._canonify(value)
+        if self._algorithm(rut) == verificador:
+            return self._format(rut, verificador)
+        else:
+            raise ValidationError(u'The Chilean RUT is not valid.')
+        
+    
+    def _algorithm(self, rut):
+        """
+        Takes RUT in pure canonical form, calculates the verifier digit. 
+        """
+        suma  = 0
+        multi = 2
+        for r in rut[::-1]:
+            suma  += int (r) * multi
+            multi += 1
+            if multi == 8:
+                multi = 2
+        rest = suma % 11
+        rest = 11 - rest
+        if rest == 11: return '0'
+        elif rest == 10: return 'K'
+        else: return str(rest)
+
+    def _canonify(self, rut):
+        """
+        Turns the rut into one normalized format. Returns a (rut, verifier) tuple.
+        """
+        rut = str(rut)
+        rut = rut.replace(' ', '')
+        rut = rut.replace('.', '')
+        rut = rut.replace('-', '')
+        return (rut[:-1], rut[-1])
+
+    def _format(self, code, verifier=None):
+        """
+        Formats the rut from canonical form to the common string representation.
+        If verifier=None, then the last digit in 'code' is the verifier.
+        """
+        if(verifier == None):
+            verifier = code[-1]
+            code = code[:-1]
+        while len(code) > 3 and (code.find('.') > 3 or code.find('.') == -1):
+            if code.find('.') == -1:
+                new_dot = -3
+            else:
+                new_dot = code.find('.')-3
+            code = code[:new_dot] +'.'+code[new_dot:]    
+        return '%s-%s' % (code, verifier)
+

tests/regressiontests/forms/localflavor.py

@@ -1052 +1052 @@
 <option value="VIC">Victoria</option>
 <option value="WA">Western Australia</option>
 </select>
+
+
+## CLRutField #############################################################
+
+CLRutField is a Field that checks the validity of the Chilean
+personal identification number (RUT). It has two modes relaxed (default) and
+strict. 
+
+>>> from django.contrib.localflavor.cl.forms import CLRutField
+>>> rut = CLRutField()
+
+>>> rut.clean('11-6')
+'11-6'
+>>> rut.clean('116')
+'11-6'
+
+# valid format, bad verifier.
+>>> rut.clean('11.111.111-0')
+Traceback (most recent call last):
+...
+ValidationError: [u'The Chilean RUT is not valid.']
+>>> rut.clean('111')
+Traceback (most recent call last):
+...
+ValidationError: [u'The Chilean RUT is not valid.']
+
+>>> rut.clean('767484100')
+'76.748.410-0'
+>>> rut.clean('78.412.790-7')
+'78.412.790-7'
+>>> rut.clean('8.334.6043')
+'8.334.604-3'
+>>> rut.clean('76793310-K')
+'76.793.310-K'
+
+Strict RUT usage (does not allow imposible values)
+>>> rut = CLRutField(strict=True)
+
+>>> rut.clean('11-6')
+Traceback (most recent call last):
+...
+ValidationError: [u'Enter valid a Chilean Rut. The format is XX.XXX.XXX-X.']
+
+# valid format, bad verifier.
+>>> rut.clean('11.111.111-0')
+Traceback (most recent call last):
+...
+ValidationError: [u'The Chilean RUT is not valid.']
+
+# Correct input, invalid format.
+>>> rut.clean('767484100')
+Traceback (most recent call last):
+...
+ValidationError: [u'Enter valid a Chilean Rut. The format is XX.XXX.XXX-X.']
+>>> rut.clean('78.412.790-7')
+'78.412.790-7'
+>>> rut.clean('8.334.6043')
+Traceback (most recent call last):
+...
+ValidationError: [u'Enter valid a Chilean Rut. The format is XX.XXX.XXX-X.']
+>>> rut.clean('76793310-K')
+Traceback (most recent call last):
+...
+ValidationError: [u'Enter valid a Chilean Rut. The format is XX.XXX.XXX-X.']
+
 """