Ticket #11517: patch_clpatentefield.diff

django/contrib/localflavor/cl/forms.py

@@ -3 +3 @@
 """
 
 from django.forms import ValidationError
-from django.forms.fields import RegexField, Select, EMPTY_VALUES
+from django.forms.fields import RegexField, Select, EMPTY_VALUES, CharField
 from django.utils.translation import ugettext_lazy as _
 from django.utils.encoding import smart_unicode
 
+OLD = 0
+NEW = 1
 
+
 class CLRegionSelect(Select):
     """
     A Select widget that uses a list of Chilean Regions (Regiones)
     as its choices.
     """
+
     def __init__(self, attrs=None):
         from cl_regions import REGION_CHOICES
         super(CLRegionSelect, self).__init__(attrs, choices=REGION_CHOICES)
 
+
 class CLRutField(RegexField):
     """
     Chilean "Rol Unico Tributario" (RUT) field. This is the Chilean national
@@ -92 +97 @@
             code = code[:new_dot] + '.' + code[new_dot:]
         return u'%s-%s' % (code, verifier)
 
+
+class CLPatenteField(CharField):
+    """
+    Chilean "Patente", Placa Patente Field, support old LL-NNNN and new LLLL-NN
+    type, where L in new can't be vowel or m,n and q.
+    """
+
+    default_error_messages = CharField.default_error_messages
+    default_error_messages.update({'invalid': _('Enter a valid Chilean Car Plate.'),})
+
+    def __init__(self, *args, **kwargs):
+        kwargs['max_length'] = 7
+        kwargs['min_length'] = 6
+        kwargs['label'] = "Valor"
+        super(CLPatenteField, self).__init__(*args, **kwargs)
+
+    def clean(self, value):
+        """
+        Check and clean the field for length and type.
+        """
+        super(CLPatenteField, self).clean(value)
+        if value in EMPTY_VALUES:
+            return u''
+
+        value = smart_unicode(value.lower())
+        value_length = len(value)
+        if self.max_length is not None and value_length > self.max_length:
+            raise ValidationError(self.error_messages['max_length'] % {'max': self.max_length, 'length': value_length})
+        if self.min_length is not None and value_length < self.min_length:
+            raise ValidationError(self.error_messages['min_length'] % {'min': self.min_length, 'length': value_length})
+
+        value = self._canonify(value)
+        _type = self._algorithm(value)
+
+        if _type not in (OLD, NEW):
+            raise ValidationError(self.error_messages['invalid'])
+
+        return self._format(value.upper(), _type)
+
+    def _algorithm(self, value):
+        """
+        Takes Car Plate in canonical form and verify if it is valid.
+        Returns the type or None it is not valid.
+        """
+        import re
+        # Old type patern
+        op = '^[a-z]{2,2}\d{4,4}$'
+        op = re.compile(op)
+        # New type patern
+        np = '^[bcdfghjklprstvwxyz]{4,4}\d{2,2}$'
+        np = re.compile(np)
+        if re.search(op, value):
+            return OLD
+        elif re.search(np, value):
+            return NEW
+        else:
+            return None
+
+    def _canonify(self, patente):
+        """
+        Turns the Car Plate into one normalized format.
+        """
+        return patente.replace(' ', '').replace('-', '')
+
+    def _format(self, value, _type):
+        """
+        Formats the Car Plate from canonical form to the common string
+        representation. LL-DDDD or LLLL-DD
+        """
+        if _type == OLD:
+            return u'-'.join((value[:2], value[2:]))
+        if _type == NEW:
+            return u'-'.join((value[:4], value[4:]))