Ticket #8735: django_localflavor_pt.diff

django/contrib/localflavor/pt/forms.py

@@ +1 @@
+"""
+PT-specific Form helpers
+"""
+
+from django.forms import ValidationError
+from django.forms.fields import Field, RegexField, Select, EMPTY_VALUES
+from django.utils.encoding import smart_unicode
+from django.utils.translation import ugettext_lazy as _
+import re
+
+phone_digits_re = re.compile(r'^(\d{9}|(00|\+)\d*)$')
+
+
+class PTZipCodeField(RegexField):
+    default_error_messages = {
+        'invalid': _('Enter a zip code in the format XXXX-XXX.'),
+    }
+
+    def __init__(self, *args, **kwargs):
+        super(PTZipCodeField, self).__init__(r'^(\d{4}-\d{3}|\d{7})$',
+            max_length=None, min_length=None, *args, **kwargs)
+
+    def clean(self,value):
+        cleaned = super(PTZipCodeField, self).clean(value)
+        if len(cleaned) == 7:
+               return u'%s-%s' % (cleaned[:4],cleaned[4:])
+        else:
+               return cleaned
+                
+class PTPhoneNumberField(Field):
+    """
+    Validate local Portuguese phone number (including international ones)
+        It should have 9 digits (may include spaces) or start by 00 or + (international)
+    """
+    default_error_messages = {
+        'invalid': u'Phone numbers must have 9 digits, or start by + or 00.',
+    }
+
+    def clean(self, value):
+        super(PTPhoneNumberField, self).clean(value)
+        if value in EMPTY_VALUES:
+            return u''
+        value = re.sub('(\.|\s)', '', smart_unicode(value))
+        m = phone_digits_re.search(value)
+        if m:
+            return u'%s' % value
+        raise ValidationError(self.error_messages['invalid'])
+

tests/regressiontests/forms/localflavor/pt.py

@@ +1 @@
+# -*- coding: utf-8 -*-
+# Tests for the contrib/localflavor/ PT form fields.
+
+tests = r"""
+# PTZipCodeField #############################################################
+
+PTZipCodeField validates that the data is a valid PT zipcode.
+>>> from django.contrib.localflavor.pt.forms import PTZipCodeField
+>>> f = PTZipCodeField()
+>>> f.clean('3030-034')
+u'3030-034'
+>>> f.clean('1003456')
+u'1003-456'
+>>> f.clean('2A200')
+Traceback (most recent call last):
+...
+ValidationError: [u'Enter a zip code in the format XXXX-XXX.']
+>>> f.clean('980001')
+Traceback (most recent call last):
+...
+ValidationError: [u'Enter a zip code in the format XXXX-XXX.']
+>>> f.clean(None)
+Traceback (most recent call last):
+...
+ValidationError: [u'This field is required.']
+>>> f.clean('')
+Traceback (most recent call last):
+...
+ValidationError: [u'This field is required.']
+
+>>> f = PTZipCodeField(required=False)
+>>> f.clean('3030-034')
+u'3030-034'
+>>> f.clean('1003456')
+u'1003-456'
+>>> f.clean('2A200')
+Traceback (most recent call last):
+...
+ValidationError: [u'Enter a zip code in the format XXXX-XXX.']
+>>> f.clean('980001')
+Traceback (most recent call last):
+...
+ValidationError: [u'Enter a zip code in the format XXXX-XXX.']
+>>> f.clean(None)
+u''
+>>> f.clean('')
+u''
+
+# PTPhoneNumberField ##########################################################
+
+PTPhoneNumberField validates that the data is a valid Portuguese phone number.
+It's normalized to XXXXXXXXX format  or +X(X) for international numbers. Dots are valid too.
+>>> from django.contrib.localflavor.pt.forms import PTPhoneNumberField
+>>> f = PTPhoneNumberField()
+>>> f.clean('917845189')
+u'917845189'
+>>> f.clean('91 784 5189')
+u'917845189'
+>>> f.clean('91 784 5189')
+u'917845189'
+>>> f.clean('+351 91 111')
+u'+35191111'
+>>> f.clean('00351873')
+u'00351873'
+>>> f.clean('91 784 51 8')
+Traceback (most recent call last):
+...
+ValidationError: [u'Phone numbers must have 9 digits, or start by + or 00.']
+>>> f.clean('091 456 987 1')
+Traceback (most recent call last):
+...
+ValidationError: [u'Phone numbers must have 9 digits, or start by + or 00.']
+>>> f.clean(None)
+Traceback (most recent call last):
+...
+ValidationError: [u'This field is required.']
+>>> f.clean('')
+Traceback (most recent call last):
+...
+ValidationError: [u'This field is required.']
+
+>>> f = PTPhoneNumberField(required=False)
+>>> f.clean('917845189')
+u'917845189'
+>>> f.clean('91 784 5189')
+u'917845189'
+>>> f.clean('91 784 5189')
+u'917845189'
+>>> f.clean('+351 91 111')
+u'+35191111'
+>>> f.clean('00351873')
+u'00351873'
+>>> f.clean('91 784 51 8')
+Traceback (most recent call last):
+...
+ValidationError: [u'Phone numbers must have 9 digits, or start by + or 00.']
+>>> f.clean('091 456 987 1')
+Traceback (most recent call last):
+...
+ValidationError: [u'Phone numbers must have 9 digits, or start by + or 00.']
+>>> f.clean(None)
+u''
+>>> f.clean('')
+u''
+
+"""

tests/regressiontests/forms/tests.py

@@ -20 +20 @@
 from localflavor.jp import tests as localflavor_jp_tests
 from localflavor.nl import tests as localflavor_nl_tests
 from localflavor.pl import tests as localflavor_pl_tests
+from localflavor.pt import tests as localflavor_pt_tests
 from localflavor.ro import tests as localflavor_ro_tests
 from localflavor.sk import tests as localflavor_sk_tests
 from localflavor.uk import tests as localflavor_uk_tests
@@ -53 +54 @@
     'localflavor_jp_tests': localflavor_jp_tests,
     'localflavor_nl_tests': localflavor_nl_tests,
     'localflavor_pl_tests': localflavor_pl_tests,
+    'localflavor_pt_tests': localflavor_pt_tests,
     'localflavor_ro_tests': localflavor_ro_tests,
     'localflavor_sk_tests': localflavor_sk_tests,
     'localflavor_uk_tests': localflavor_uk_tests,