123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
"""
|
|
Swiss-specific Form helpers
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import re
|
|
|
|
from django.contrib.localflavor.ch.ch_states import STATE_CHOICES
|
|
from django.core.validators import EMPTY_VALUES
|
|
from django.forms import ValidationError
|
|
from django.forms.fields import Field, RegexField, Select
|
|
from django.utils.encoding import smart_unicode
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
|
|
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, max_length=None, min_length=None, *args, **kwargs):
|
|
super(CHZipCodeField, self).__init__(r'^\d{4}$',
|
|
max_length, min_length, *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):
|
|
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
|
|
|
|
"""
|
|
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)
|
|
|