# -*- coding: utf-8 -*-
"""
Romanian specific form helpers.
"""
from __future__ import absolute_import, unicode_literals
import datetime
import re
from django.core.validators import EMPTY_VALUES
from django.forms import ValidationError, Field, RegexField, Select
from django.utils.translation import ugettext_lazy as _
from ..generic.forms import IBANFormField
from .ro_counties import COUNTIES_CHOICES
phone_digits_re = re.compile(r'^[0-9\-\.\(\)\s]{3,20}$')
[docs]class ROCIFField(RegexField):
"""
A Romanian fiscal identity code (CIF) field
For CIF validation algorithm see http://www.validari.ro/cui.html
"""
default_error_messages = {
'invalid': _("Enter a valid CIF."),
}
def __init__(self, max_length=10, min_length=2, *args, **kwargs):
super(ROCIFField, self).__init__(r'^(RO)?[0-9]{2,10}', max_length,
min_length, *args, **kwargs)
[docs] def clean(self, value):
"""
CIF validation
"""
value = super(ROCIFField, self).clean(value)
if value in EMPTY_VALUES:
return ''
# strip RO part
if value[0:2] == 'RO':
value = value[2:]
key = '753217532'[::-1]
value = value[::-1]
key_iter = iter(key)
checksum = 0
for digit in value[1:]:
checksum += int(digit) * int(next(key_iter))
checksum = checksum * 10 % 11
if checksum == 10:
checksum = 0
if checksum != int(value[0]):
raise ValidationError(self.error_messages['invalid'])
return value[::-1]
[docs]class ROCNPField(RegexField):
"""
A Romanian personal identity code (CNP) field
For CNP validation algorithm see http://www.validari.ro/cnp.html
"""
default_error_messages = {
'invalid': _("Enter a valid CNP."),
}
def __init__(self, max_length=13, min_length=13, *args, **kwargs):
super(ROCNPField, self).__init__(r'^[1-9][0-9]{12}', max_length,
min_length, *args, **kwargs)
[docs] def clean(self, value):
"""
CNP validations
"""
value = super(ROCNPField, self).clean(value)
if value in EMPTY_VALUES:
return ''
# check birthdate digits
try:
datetime.date(int(value[1:3]), int(value[3:5]), int(value[5:7]))
except ValueError:
raise ValidationError(self.error_messages['invalid'])
# checksum
key = '279146358279'
checksum = 0
value_iter = iter(value)
for digit in key:
checksum += int(digit) * int(next(value_iter))
checksum %= 11
if checksum == 10:
checksum = 1
if checksum != int(value[12]):
raise ValidationError(self.error_messages['invalid'])
return value
[docs]class ROCountyField(Field):
"""
A form field that validates its input is a Romanian county name or
abbreviation. It normalizes the input to the standard vehicle registration
abbreviation for the given county
WARNING: This field will only accept names written with diacritics; consider
using ROCountySelect if this behavior is unnaceptable for you
Example:
Argeş => valid
Arges => invalid
"""
default_error_messages = {
'invalid': 'Enter a Romanian county code or name.',
}
def clean(self, value):
super(ROCountyField, self).clean(value)
if value in EMPTY_VALUES:
return ''
try:
value = value.strip().upper()
except AttributeError:
pass
# search for county code
for entry in COUNTIES_CHOICES:
if value in entry:
return value
# search for county name
normalized_CC = []
for entry in COUNTIES_CHOICES:
normalized_CC.append((entry[0], entry[1].upper()))
for entry in normalized_CC:
if entry[1] == value:
return entry[0]
raise ValidationError(self.error_messages['invalid'])
[docs]class ROCountySelect(Select):
"""
A Select widget that uses a list of Romanian counties (judete) as its
choices.
"""
def __init__(self, attrs=None):
super(ROCountySelect, self).__init__(attrs, choices=COUNTIES_CHOICES)
[docs]class ROIBANField(IBANFormField):
"""
Romanian International Bank Account Number (IBAN) field
.. versionchanged:: 1.1
Validation error messages changed to the messages used in :class:`.IBANFormField`
.. deprecated:: 1.1
Use `IBANFormField` with `included_countries=('RO',)` option instead.
"""
def __init__(self, *args, **kwargs):
super(ROIBANField, self).__init__(use_nordea_extensions=False, include_countries=('RO',), **kwargs)
[docs]class ROPhoneNumberField(RegexField):
"""
Romanian phone number field
.. versionchanged:: 1.1
Made the field also accept national short phone numbers and 7-digit
regional phone numbers besides the regular ones.
Official documentation (in English): http://www.ancom.org.ro/en/pnn_1300
Official documentation (in Romanian): http://www.ancom.org.ro/pnn_1300
"""
default_error_messages = {
'invalid_length':
_('Phone numbers may only have 7 or 10 digits, except the ' +
'national short numbers which have 3 to 6 digits'),
'invalid_long_format':
_('Normal phone numbers (7 or 10 digits) must begin with "0"'),
'invalid_short_format':
_('National short numbers (3 to 6 digits) must begin with "1"'),
}
def __init__(self, max_length=20, min_length=3, *args, **kwargs):
super(ROPhoneNumberField, self).__init__(phone_digits_re,
max_length, min_length, *args, **kwargs)
[docs] def clean(self, value):
"""
Strips braces, dashes, dots and spaces. Checks the final length.
"""
value = super(ROPhoneNumberField, self).clean(value)
if value in EMPTY_VALUES:
return ''
value = re.sub('[()-. ]', '', value)
length = len(value)
if length in (3, 4, 5, 6, 7, 10):
if (length == 7 or length == 10) and value[0] != '0':
raise ValidationError(
self.error_messages['invalid_long_format'])
elif (3 <= length and length <= 6) and value[0] != '1':
raise ValidationError(
self.error_messages['invalid_short_format'])
else:
raise ValidationError(self.error_messages['invalid_length'])
return value
[docs]class ROPostalCodeField(RegexField):
"""Romanian postal code field."""
default_error_messages = {
'invalid': _('Enter a valid postal code in the format XXXXXX'),
}
def __init__(self, max_length=6, min_length=6, *args, **kwargs):
super(ROPostalCodeField, self).__init__(r'^[0-9][0-8][0-9]{4}$',
max_length, min_length, *args, **kwargs)