Added finnish ssn validator

kvesteri · kvesteri · commit 05c35b05f66c · 2013-10-29T13:03:37.000+02:00
diff --git a/validators/__init__.py b/validators/__init__.py
@@ -1,20 +1,21 @@
 from .between import between
 from .email import email
 from .extremes import Min, Max
-from .finnish_business_id import finnish_business_id
 from .ip_address import ipv4, ipv6
 from .length import length
 from .mac_address import mac_address
 from .truthy import truthy
 from .url import url
 from .utils import ValidationFailure, validator
 from .uuid import uuid
+from .i18n import finnish_business_id, finnish_ssn
 
 
 __all__ = (
     between,
     email,
     finnish_business_id,
+    finnish_ssn,
     ipv4,
     ipv6,
     length,
diff --git a/validators/email.py b/validators/email.py
@@ -8,7 +8,7 @@
     r"(\.[-!#$%&'*+/=?^_`{}|~0-9A-Z]+)*$"
     # quoted-string
     r'|^"([\001-\010\013\014\016-\037!#-\[\]-\177]|'
-    r'\\[\001-\011\013\014\016-\177])*"$)',
+    r"""\\[\001-\011\013\014\016-\177])*"$)""",
     re.IGNORECASE
 )
 domain_regex = re.compile(
diff --git a/validators/finnish_business_id.py b/validators/finnish_business_id.py
diff --git a/validators/i18n/__init__.py b/validators/i18n/__init__.py
@@ -0,0 +1,7 @@
+from .fi import finnish_ssn, finnish_business_id
+
+
+__all__ = (
+    finnish_ssn,
+    finnish_business_id,
+)
diff --git a/validators/i18n/fi.py b/validators/i18n/fi.py
@@ -0,0 +1,72 @@
+import re
+from validators.utils import validator
+
+
+business_id_pattern = re.compile(r'^[0-9]{7}-[0-9]$')
+ssn_checkmarks = '0123456789ABCDEFHJKLMNPRSTUVWXY'
+ssn_pattern = re.compile(
+    r"""^
+    (?P<date>([0-2]\d|3[01])
+    (0\d|1[012])
+    (\d{{2}}))
+    [A+-]
+    (?P<serial>(\d{{3}}))
+    (?P<checksum>[{checkmarks}])$""".format(checkmarks=ssn_checkmarks),
+    re.VERBOSE | re.IGNORECASE
+)
+
+
+@validator
+def finnish_business_id(business_id):
+    """
+    Validates a Finnish Business ID. Each company in Finland has a distinct
+    business id. For more information see `Finnish Trade Register`_
+
+    .. _Finnish Trade Register:
+        http://en.wikipedia.org/wiki/Finnish_Trade_Register
+
+    ::
+
+        >>> assert finnish_business_id('0112038-9')  # Fast Monkeys Ltd
+
+        >>> assert not finnish_business_id('1234567-8')  # Bogus ID
+
+    .. versionadded:: 0.4
+
+    :param business_id: business_id to validate
+    """
+    if not business_id or not re.match(business_id_pattern, business_id):
+        return False
+    factors = [7, 9, 10, 5, 8, 4, 2]
+    numbers = map(int, business_id[:7])
+    checksum = int(business_id[8])
+    sum_ = sum(f * n for f, n in zip(factors, numbers))
+    modulo = sum_ % 11
+    return (11 - modulo == checksum) or (modulo == 0 and checksum == 0)
+
+
+@validator
+def finnish_ssn(ssn):
+    """
+    Validates a Finnish Social Security Number. This validator is based on
+    `django-localflavor-fi`_.
+
+    .. _django-localflavor-fi:
+        https://github.com/django/django-localflavor-fi/
+
+    .. versionadded:: 0.5
+
+    :param ssn: Social Security Number to validate
+    """
+    if not ssn:
+        return False
+
+    result = re.match(ssn_pattern, ssn)
+    if not result:
+        return False
+    gd = result.groupdict()
+    checksum = int(gd['date'] + gd['serial'])
+    return (
+        ssn_checkmarks[checksum % len(ssn_checkmarks)] ==
+        gd['checksum'].upper()
+    )

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@`
`8`	`8`	r"(\.[-!#$%&'+/=?^_`{}\|~0-9A-Z]+)$"
`9`	`9`	`# quoted-string`
`10`	`10`	`r'\|^"([\001-\010\013\014\016-\037!#-\[\]-\177]\|'`
`11`		`- r'\\[\001-\011\013\014\016-\177])*"$)',`
	`11`	`+ r"""\\[\001-\011\013\014\016-\177])*"$)""",`
`12`	`12`	`re.IGNORECASE`
`13`	`13`	`)`
`14`	`14`	`domain_regex = re.compile(`