You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.8 KiB
118 lines
3.8 KiB
""" |
|
Swiss-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 |
|
|
|
id_re = re.compile(r"^(?P<idnumber>\w{8})(?P<pos9>(\d{1}|<))(?P<checksum>\d{1})$") |
|
phone_digits_re = re.compile(r'^0([1-9]{1})\d{8}$') |
|
|
|
class CHZipCodeField(RegexField): |
|
default_error_messages = { |
|
'invalid': _('Enter a zip code in the format XXXX.'), |
|
} |
|
|
|
def __init__(self, *args, **kwargs): |
|
super(CHZipCodeField, self).__init__(r'^\d{4}$', |
|
max_length=None, min_length=None, *args, **kwargs) |
|
|
|
class CHPhoneNumberField(Field): |
|
""" |
|
Validate local Swiss phone number (not international ones) |
|
The correct format is '0XX XXX XX XX'. |
|
'0XX.XXX.XX.XX' and '0XXXXXXXXX' validate but are corrected to |
|
'0XX XXX XX XX'. |
|
""" |
|
default_error_messages = { |
|
'invalid': 'Phone numbers must be in 0XX XXX XX XX format.', |
|
} |
|
|
|
def clean(self, value): |
|
super(CHPhoneNumberField, 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 %s %s %s' % (value[0:3], value[3:6], value[6:8], value[8:10]) |
|
raise ValidationError(self.error_messages['invalid']) |
|
|
|
class CHStateSelect(Select): |
|
""" |
|
A Select widget that uses a list of CH states as its choices. |
|
""" |
|
def __init__(self, attrs=None): |
|
from ch_states import STATE_CHOICES # relative import |
|
super(CHStateSelect, self).__init__(attrs, choices=STATE_CHOICES) |
|
|
|
class CHIdentityCardNumberField(Field): |
|
""" |
|
A Swiss identity card number. |
|
|
|
Checks the following rules to determine whether the number is valid: |
|
|
|
* Conforms to the X1234567<0 or 1234567890 format. |
|
* Included checksums match calculated checksums |
|
|
|
Algorithm is documented at http://adi.kousz.ch/artikel/IDCHE.htm |
|
""" |
|
default_error_messages = { |
|
'invalid': _('Enter a valid Swiss identity or passport card number in X1234567<0 or 1234567890 format.'), |
|
} |
|
|
|
def has_valid_checksum(self, number): |
|
given_number, given_checksum = number[:-1], number[-1] |
|
new_number = given_number |
|
calculated_checksum = 0 |
|
fragment = "" |
|
parameter = 7 |
|
|
|
first = str(number[:1]) |
|
if first.isalpha(): |
|
num = ord(first.upper()) - 65 |
|
if num < 0 or num > 8: |
|
return False |
|
new_number = str(num) + new_number[1:] |
|
new_number = new_number[:8] + '0' |
|
|
|
if not new_number.isdigit(): |
|
return False |
|
|
|
for i in range(len(new_number)): |
|
fragment = int(new_number[i])*parameter |
|
calculated_checksum += fragment |
|
|
|
if parameter == 1: |
|
parameter = 7 |
|
elif parameter == 3: |
|
parameter = 1 |
|
elif parameter ==7: |
|
parameter = 3 |
|
|
|
return str(calculated_checksum)[-1] == given_checksum |
|
|
|
def clean(self, value): |
|
super(CHIdentityCardNumberField, self).clean(value) |
|
if value in EMPTY_VALUES: |
|
return u'' |
|
|
|
match = re.match(id_re, value) |
|
if not match: |
|
raise ValidationError(self.error_messages['invalid']) |
|
|
|
idnumber, pos9, checksum = match.groupdict()['idnumber'], match.groupdict()['pos9'], match.groupdict()['checksum'] |
|
|
|
if idnumber == '00000000' or \ |
|
idnumber == 'A0000000': |
|
raise ValidationError(self.error_messages['invalid']) |
|
|
|
all_digits = "%s%s%s" % (idnumber, pos9, checksum) |
|
if not self.has_valid_checksum(all_digits): |
|
raise ValidationError(self.error_messages['invalid']) |
|
|
|
return u'%s%s%s' % (idnumber, pos9, checksum) |
|
|
|
|