Django-1.2.1/django/contrib/localflavor/cz/forms.py

   1 """
   2 Czech-specific form helpers
   3 """
   4
   5 from django.core.validators import EMPTY_VALUES
   6 from django.forms import ValidationError
   7 from django.forms.fields import Select, RegexField, Field
   8 from django.utils.translation import ugettext_lazy as _
   9 import re
  10
  11 birth_number = re.compile(r'^(?P<birth>\d{6})/?(?P<id>\d{3,4})$')
  12 ic_number = re.compile(r'^(?P<number>\d{7})(?P<check>\d)$')
  13
  14 class CZRegionSelect(Select):
  15     """
  16     A select widget widget with list of Czech regions as choices.
  17     """
  18     def __init__(self, attrs=None):
  19         from cz_regions import REGION_CHOICES
  20         super(CZRegionSelect, self).__init__(attrs, choices=REGION_CHOICES)
  21
  22 class CZPostalCodeField(RegexField):
  23     """
  24     A form field that validates its input as Czech postal code.
  25     Valid form is XXXXX or XXX XX, where X represents integer.
  26     """
  27     default_error_messages = {
  28         'invalid': _(u'Enter a postal code in the format XXXXX or XXX XX.'),
  29     }
  30
  31     def __init__(self, *args, **kwargs):
  32         super(CZPostalCodeField, self).__init__(r'^\d{5}$|^\d{3} \d{2}$',
  33             max_length=None, min_length=None, *args, **kwargs)
  34
  35     def clean(self, value):
  36         """
  37         Validates the input and returns a string that contains only numbers.
  38         Returns an empty string for empty values.
  39         """
  40         v = super(CZPostalCodeField, self).clean(value)
  41         return v.replace(' ', '')
  42
  43 class CZBirthNumberField(Field):
  44     """
  45     Czech birth number field.
  46     """
  47     default_error_messages = {
  48         'invalid_format': _(u'Enter a birth number in the format XXXXXX/XXXX or XXXXXXXXXX.'),
  49         'invalid_gender': _(u'Invalid optional parameter Gender, valid values are \'f\' and \'m\''),
  50         'invalid': _(u'Enter a valid birth number.'),
  51     }
  52
  53     def clean(self, value, gender=None):
  54         super(CZBirthNumberField, self).__init__(value)
  55
  56         if value in EMPTY_VALUES:
  57             return u''
  58
  59         match = re.match(birth_number, value)
  60         if not match:
  61             raise ValidationError(self.error_messages['invalid_format'])
  62
  63         birth, id = match.groupdict()['birth'], match.groupdict()['id']
  64
  65         # Three digits for verificatin number were used until 1. january 1954
  66         if len(id) == 3:
  67             return u'%s' % value
  68
  69         # Birth number is in format YYMMDD. Females have month value raised by 50.
  70         # In case that all possible number are already used (for given date),
  71         #  the month field is raised by 20.
  72         if gender is not None:
  73             if gender == 'f':
  74                 female_const = 50
  75             elif gender == 'm':
  76                 female_const = 0
  77             else:
  78                 raise ValidationError(self.error_messages['invalid_gender'])
  79
  80             month = int(birth[2:4]) - female_const
  81             if (not 1 <= month <= 12):
  82                 if (not 1 <= (month - 20) <= 12):
  83                     raise ValidationError(self.error_messages['invalid'])
  84
  85         day = int(birth[4:6])
  86         if not (1 <= day <= 31):
  87             raise ValidationError(self.error_messages['invalid'])
  88
  89         # Fourth digit has been added since 1. January 1954.
  90         # It is modulo of dividing birth number and verification number by 11.
  91         # If the modulo were 10, the last number was 0 (and therefore, the whole
  92         # birth number wasn't divisable by 11. These number are no longer used (since 1985)
  93         # and the condition 'modulo == 10' can be removed in 2085.
  94
  95         modulo = int(birth + id[:3]) % 11
  96
  97         if (modulo == int(id[-1])) or (modulo == 10 and id[-1] == '0'):
  98             return u'%s' % value
  99         else:
 100             raise ValidationError(self.error_messages['invalid'])
 101
 102 class CZICNumberField(Field):
 103     """
 104     Czech IC number field.
 105     """
 106     default_error_messages = {
 107         'invalid': _(u'Enter a valid IC number.'),
 108     }
 109
 110     def clean(self, value):
 111         super(CZICNumberField, self).__init__(value)
 112
 113         if value in EMPTY_VALUES:
 114             return u''
 115
 116         match = re.match(ic_number, value)
 117         if not match:
 118             raise ValidationError(self.error_messages['invalid'])
 119
 120         number, check = match.groupdict()['number'], int(match.groupdict()['check'])
 121
 122         sum = 0
 123         weight = 8
 124         for digit in number:
 125             sum += int(digit)*weight
 126             weight -= 1
 127
 128         remainder = sum % 11
 129
 130         # remainder is equal:
 131         #  0 or 10: last digit is 1
 132         #  1: last digit is 0
 133         # in other case, last digin is 11 - remainder
 134
 135         if (not remainder % 10 and check == 1) or \
 136         (remainder == 1 and check == 0) or \
 137         (check == (11 - remainder)):
 138             return u'%s' % value
 139
 140         raise ValidationError(self.error_messages['invalid'])
 141