diff -r 000000000000 -r 0d40e90630ef web/lib/django/contrib/localflavor/cz/forms.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/web/lib/django/contrib/localflavor/cz/forms.py Wed Jan 20 00:34:04 2010 +0100 @@ -0,0 +1,140 @@ +""" +Czech-specific form helpers +""" + +from django.forms import ValidationError +from django.forms.fields import Select, RegexField, Field, EMPTY_VALUES +from django.utils.translation import ugettext_lazy as _ +import re + +birth_number = re.compile(r'^(?P\d{6})/?(?P\d{3,4})$') +ic_number = re.compile(r'^(?P\d{7})(?P\d)$') + +class CZRegionSelect(Select): + """ + A select widget widget with list of Czech regions as choices. + """ + def __init__(self, attrs=None): + from cz_regions import REGION_CHOICES + super(CZRegionSelect, self).__init__(attrs, choices=REGION_CHOICES) + +class CZPostalCodeField(RegexField): + """ + A form field that validates its input as Czech postal code. + Valid form is XXXXX or XXX XX, where X represents integer. + """ + default_error_messages = { + 'invalid': _(u'Enter a postal code in the format XXXXX or XXX XX.'), + } + + def __init__(self, *args, **kwargs): + super(CZPostalCodeField, self).__init__(r'^\d{5}$|^\d{3} \d{2}$', + max_length=None, min_length=None, *args, **kwargs) + + def clean(self, value): + """ + Validates the input and returns a string that contains only numbers. + Returns an empty string for empty values. + """ + v = super(CZPostalCodeField, self).clean(value) + return v.replace(' ', '') + +class CZBirthNumberField(Field): + """ + Czech birth number field. + """ + default_error_messages = { + 'invalid_format': _(u'Enter a birth number in the format XXXXXX/XXXX or XXXXXXXXXX.'), + 'invalid_gender': _(u'Invalid optional parameter Gender, valid values are \'f\' and \'m\''), + 'invalid': _(u'Enter a valid birth number.'), + } + + def clean(self, value, gender=None): + super(CZBirthNumberField, self).__init__(value) + + if value in EMPTY_VALUES: + return u'' + + match = re.match(birth_number, value) + if not match: + raise ValidationError(self.error_messages['invalid_format']) + + birth, id = match.groupdict()['birth'], match.groupdict()['id'] + + # Three digits for verificatin number were used until 1. january 1954 + if len(id) == 3: + return u'%s' % value + + # Birth number is in format YYMMDD. Females have month value raised by 50. + # In case that all possible number are already used (for given date), + # the month field is raised by 20. + if gender is not None: + if gender == 'f': + female_const = 50 + elif gender == 'm': + female_const = 0 + else: + raise ValidationError(self.error_messages['invalid_gender']) + + month = int(birth[2:4]) - female_const + if (not 1 <= month <= 12): + if (not 1 <= (month - 20) <= 12): + raise ValidationError(self.error_messages['invalid']) + + day = int(birth[4:6]) + if not (1 <= day <= 31): + raise ValidationError(self.error_messages['invalid']) + + # Fourth digit has been added since 1. January 1954. + # It is modulo of dividing birth number and verification number by 11. + # If the modulo were 10, the last number was 0 (and therefore, the whole + # birth number wasn't divisable by 11. These number are no longer used (since 1985) + # and the condition 'modulo == 10' can be removed in 2085. + + modulo = int(birth + id[:3]) % 11 + + if (modulo == int(id[-1])) or (modulo == 10 and id[-1] == '0'): + return u'%s' % value + else: + raise ValidationError(self.error_messages['invalid']) + +class CZICNumberField(Field): + """ + Czech IC number field. + """ + default_error_messages = { + 'invalid': _(u'Enter a valid IC number.'), + } + + def clean(self, value): + super(CZICNumberField, self).__init__(value) + + if value in EMPTY_VALUES: + return u'' + + match = re.match(ic_number, value) + if not match: + raise ValidationError(self.error_messages['invalid']) + + number, check = match.groupdict()['number'], int(match.groupdict()['check']) + + sum = 0 + weight = 8 + for digit in number: + sum += int(digit)*weight + weight -= 1 + + remainder = sum % 11 + + # remainder is equal: + # 0 or 10: last digit is 1 + # 1: last digit is 0 + # in other case, last digin is 11 - remainder + + if (not remainder % 10 and check == 1) or \ + (remainder == 1 and check == 0) or \ + (check == (11 - remainder)): + return u'%s' % value + + raise ValidationError(self.error_messages['invalid']) +