web/lib/django/contrib/localflavor/pe/forms.py
changeset 0 0d40e90630ef
child 29 cc9b7e14412b
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/web/lib/django/contrib/localflavor/pe/forms.py	Wed Jan 20 00:34:04 2010 +0100
@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+"""
+PE-specific Form helpers.
+"""
+
+from django.forms import ValidationError
+from django.forms.fields import RegexField, CharField, Select, EMPTY_VALUES
+from django.utils.translation import ugettext_lazy as _
+
+class PERegionSelect(Select):
+    """
+    A Select widget that uses a list of Peruvian Regions as its choices.
+    """
+    def __init__(self, attrs=None):
+        from pe_region import REGION_CHOICES
+        super(PERegionSelect, self).__init__(attrs, choices=REGION_CHOICES)
+
+class PEDNIField(CharField):
+    """
+    A field that validates `Documento Nacional de IdentidadŽ (DNI) numbers.
+    """
+    default_error_messages = {
+        'invalid': _("This field requires only numbers."),
+        'max_digits': _("This field requires 8 digits."),
+    }
+
+    def __init__(self, *args, **kwargs):
+        super(PEDNIField, self).__init__(max_length=8, min_length=8, *args,
+                **kwargs)
+
+    def clean(self, value):
+        """
+        Value must be a string in the XXXXXXXX formats.
+        """
+        value = super(PEDNIField, self).clean(value)
+        if value in EMPTY_VALUES:
+            return u''
+        if not value.isdigit():
+            raise ValidationError(self.error_messages['invalid'])
+        if len(value) != 8:
+            raise ValidationError(self.error_messages['max_digits'])
+
+        return value
+
+class PERUCField(RegexField):
+    """
+    This field validates a RUC (Registro Unico de Contribuyentes). A RUC is of
+    the form XXXXXXXXXXX.
+    """
+    default_error_messages = {
+        'invalid': _("This field requires only numbers."),
+        'max_digits': _("This field requires 11 digits."),
+    }
+
+    def __init__(self, *args, **kwargs):
+        super(PERUCField, self).__init__(max_length=11, min_length=11, *args,
+            **kwargs)
+
+    def clean(self, value):
+        """
+        Value must be an 11-digit number.
+        """
+        value = super(PERUCField, self).clean(value)
+        if value in EMPTY_VALUES:
+            return u''
+        if not value.isdigit():
+            raise ValidationError(self.error_messages['invalid'])
+        if len(value) != 11:
+            raise ValidationError(self.error_messages['max_digits'])
+        return value
+