Ticket #3988: patch.diff

ca/ca_provinces.py

@@ +1 @@
+"""
+An alphabetical list of provinces and territories for use as `choices`
+in a formfield.
+
+Source: http://www.canada.gc.ca/othergov/prov_e.html
+
+This exists in this standalone file so that it's only imported into memory
+when explicitly needed.
+"""
+
+PROVINCE_CHOICES = (
+    ('AB', 'Alberta'),
+    ('BC', 'British Columbia'),
+    ('MB', 'Manitoba'),
+    ('NB', 'New Brunswick'),
+    ('NF', 'Newfoundland and Labrador'),
+    ('NT', 'Northwest Territories'),
+    ('NS', 'Nova Scotia'),
+    ('NU', 'Nunavut'),
+    ('ON', 'Ontario'),
+    ('PE', 'Prince Edward Island'),
+    ('QC', 'Quebec'),
+    ('SK', 'Saskatchewan'),
+    ('YK', 'Yukon')
+)

ca/forms.py

@@ +1 @@
+"""
+Canada-specific Form helpers
+"""
+
+from django.newforms import ValidationError
+from django.newforms.fields import Field, RegexField, Select, EMPTY_VALUES
+from django.newforms.util import smart_unicode
+from django.utils.translation import gettext
+import re
+
+PHONE_DIGITS_RE = re.compile(r'^(\d{10})$')
+
+class CAPostCodeField(RegexField):
+    """Canadian post code field."""
+    def __init__(self, *args, **kwargs):
+        super(CAPostCodeField, self).__init__(r'^[A-Z]\d[A-Z] \d[A-Z]\d$',
+            max_length=None, min_length=None,
+            error_message=gettext(u'Enter correct postal code.'),
+            *args, **kwargs)
+
+class CAPhoneNumberField(Field):
+    """Canadian phone number field."""
+    def clean(self, value):
+        """Validate a phone number. Strips parentheses, whitespace and
+        hyphens.
+        """
+        super(CAPhoneNumberField, self).clean(value)
+        if value in EMPTY_VALUES:
+            return u''
+        value = re.sub('(\(|\)|\s+|-)', '', smart_unicode(value))
+        phone_match = PHONE_DIGITS_RE.search(value)
+        if phone_match:
+            return u'%s' % phone_match.group(1)
+        raise ValidationError(u'Phone numbers must contain 10 digits.')
+
+class CAProvinceSelect(Select):
+    """
+    A Select widget that uses a list of Canadian provinces and
+    territories as its choices.
+    """
+    def __init__(self, attrs=None):
+        from ca_provinces import PROVINCE_CHOICES # relative import
+        super(CAStateSelect, self).__init__(attrs, choices=PROVINCE_CHOICES)