Ticket #9042: localflavor_mx2.diff

django/contrib/localflavor/mx/forms.py

@@ +1 @@
+# -*- coding: utf-8 -*-
+
 """
 Mexican-specific form helpers.
 """
@@ -2 +4 @@
 
-from django.forms.fields import Select
+from datetime import date
 
+from django.forms import ValidationError
+from django.forms.fields import Select, RegexField, EMPTY_VALUES
+from django.utils.translation import ugettext_lazy as _
+
 class MXStateSelect(Select):
@@ -12 +18 @@
         from mx_states import STATE_CHOICES
         super(MXStateSelect, self).__init__(attrs, choices=STATE_CHOICES)
 
+class MXPostalCodeField(RegexField):
+    default_error_messages = {
+        'invalid': _('Enter a postal code in the format XXXXX.'),
+    }
+    
+    def __init__(self, *args, **kwargs):
+        super(MXPostalCodeField, self).__init__(r'^\d{5}$',
+                                                min_length=5, max_length=5,
+                                                *args, **kwargs)
+            
+class MXRFCField(RegexField):
+    """
+    Registro Federal de Contribuyente.
+    """
+    default_error_messages = {
+        'invalid': _('Invalid RFC.'),
+    }
+    
+    def __init__(self, *args, **kwargs):
+        super(MXRFCField, self).__init__(r'^[A-Z&]{3,4}\d{6}[A-Z\d]{3}$',
+                                         min_length=12, max_length=13,
+                                         *args, **kwargs)
+            
+    def clean(self, value):
+        value = super(MXRFCField, self).clean(value.upper().strip())
+        
+        if value in EMPTY_VALUES:
+            return u''
+            
+        if value[-1] != self._checksum(value[:-1]):
+            raise ValidationError(self.error_messages['invalid'])
+            
+        return value
+        
+    def _checksum(self, rfc):
+        chars = '0123456789ABCDEFGHIJKLMN&OPQRSTUVWXYZ-'
+        
+        if len(rfc) == 11:
+            rfc = '-' + rfc
+            
+        s = sum(i * chars.index(c) for i, c in zip(reversed(xrange(14)), rfc))
+        checksum = 11 - s % 11
+        
+        if checksum == 10:
+            return 'A'
+        elif checksum == 11:
+            return '0'
+        else:
+            return str(checksum)
+            
+class MXCURPField(RegexField):
+    """
+    Clave única de registro de población.
+    """
+    default_error_messages = {
+        'invalid': _('Invalid CURP.'),
+    }
+    
+    def __init__(self, *args, **kwargs):
+        super(MXCURPField, self).__init__(
+            r'^[A-Z]{4}\d{6}[HM][A-Z]{2}[B-DF-HJ-NP-TV-Z]{3}[0-9A-Z]\d$',
+            min_length=18, max_length=18, *args, **kwargs)
+            
+    def clean(self, value):
+        value = super(MXCURPField, self).clean(value.upper())
+        
+        if value in EMPTY_VALUES:
+            return u''
+        
+        try:
+            a = int(value[4:6])
+            m = int(value[6:8])
+            d = int(value[8:10])
+            a += 1900 if value[-2].isdigit() else 2000
+            date(a, m, d)
+        except: 
+            raise ValidationError(self.error_messages['invalid'])
+        
+        if value[-1] != self._checksum(value[:-1]):
+            raise ValidationError(self.error_messages['invalid'])
+            
+        return value
+        
+    def _checksum(self, curp):
+        chars = '0123456789ABCDEFGHIJKLMN&OPQRSTUVWXYZ'
+        
+        s = sum(i * chars.index(c) for i, c in zip(reversed(xrange(19)), curp))
+        checksum = 10 - s % 10
+        
+        if checksum == 10:
+            return '0'
+        else:
+            return str(checksum)

AUTHORS

@@ -395 +395 @@
     tt@gurgle.no
     David Tulig <david.tulig@gmail.com>
     Amit Upadhyay <http://www.amitu.com/blog/>
+    Marco Antonio Valenzuela Escárcega
     Geert Vanderkelen
     I.S. van Oostveen <v.oostveen@idca.nl>
     viestards.lists@gmail.com