Ticket #9042: localflavor_mx.diff

django/contrib/localflavor/mx/forms.py

@@ +1 @@
+# -*- coding: utf-8 -*-
+
 """
 Mexican-specific form helpers.
 """
@@ -2 +4 @@
 
-from django.forms.fields import Select
+import re
 
+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__(
+            re.compile(r'^[A-Z&\d]{3,4}\d{6}[A-Z\d]{3}$', re.I),
+            min_length=12, max_length=13, *args, **kwargs)
+
+    def clean(self, value):
+        value = super(MXRFCField, self).clean(value)
+        
+        if value in EMPTY_VALUES:
+            return u''
+            
+        value = value.upper().strip()
+        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):
+        curp_re = r'^[A-Z]{4}\d{6}[HM][A-Z]{2}[B-DF-HJ-NP-TV-Z]{3}[0-9A-Z]\d$'
+        super(MXCURPField, self).__init__(re.compile(curp_re, re.I),
+            min_length=18, max_length=18, *args, **kwargs)
+
+    def clean(self, value):
+        value = super(MXCURPField, self).clean(value)
+        
+        if value in EMPTY_VALUES:
+            return u''
+            
+        value = value.upper()
+        if value[-1] != self._checksum(value[:-1]):
+            raise ValidationError(self.error_messages['invalid'])
+            
+        return value
+
+    def _checksum(self, curp):
+        chars = u'0123456789ABCDEFGHIJKLMNÑOPQRSTUVWXYZ'
+        s = sum(i * chars.index(c) for i, c in zip(reversed(xrange(19)), curp))
+        checksum = 10 - (s % 10)
+        return '0' if checksum == 10 else 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 <http://cidia.com.mx>
     Geert Vanderkelen
     I.S. van Oostveen <v.oostveen@idca.nl>
     viestards.lists@gmail.com