70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
|
import re
|
||
|
|
|
||
|
|
from itertools import zip_longest
|
||
|
|
from stdnum import iso11649, luhn
|
||
|
|
from stdnum.iso7064 import mod_97_10
|
||
|
|
|
||
|
|
|
||
|
|
def sanitize_structured_reference(reference):
|
||
|
|
"""Removes whitespace and specific characters from Belgian structured references:
|
||
|
|
|
||
|
|
Example: ` RF18 1234 5678 9 ` -> `RF18123456789`
|
||
|
|
`+++020/3430/57642+++` -> `020343057642`
|
||
|
|
`***020/3430/57642***` -> `020343057642`
|
||
|
|
"""
|
||
|
|
ref = re.sub(r'\s', '', reference)
|
||
|
|
if re.match(r'(\+{3}|\*{3})\d{3}/\d{4}/\d{5}(\+{3}|\*{3})', ref):
|
||
|
|
return re.sub(r'[+*/]', '', ref)
|
||
|
|
return ref
|
||
|
|
|
||
|
|
def format_structured_reference_iso(number):
|
||
|
|
"""Format a string into a Structured Creditor Reference.
|
||
|
|
|
||
|
|
The Creditor Reference is an international standard (ISO 11649).
|
||
|
|
Example: `123456789` -> `RF18 1234 5678 9`
|
||
|
|
"""
|
||
|
|
check_digits = mod_97_10.calc_check_digits('{}RF'.format(number))
|
||
|
|
return 'RF{} {}'.format(
|
||
|
|
check_digits,
|
||
|
|
' '.join(''.join(x) for x in zip_longest(*[iter(str(number))]*4, fillvalue=''))
|
||
|
|
)
|
||
|
|
|
||
|
|
def is_valid_structured_reference_iso(reference):
|
||
|
|
"""Check whether the provided reference is a valid Structured Creditor Reference (ISO).
|
||
|
|
|
||
|
|
:param reference: the reference to check
|
||
|
|
"""
|
||
|
|
ref = sanitize_structured_reference(reference)
|
||
|
|
return iso11649.is_valid(ref)
|
||
|
|
|
||
|
|
def is_valid_structured_reference_be(reference):
|
||
|
|
"""Check whether the provided reference is a valid structured reference for Belgium.
|
||
|
|
|
||
|
|
:param reference: the reference to check
|
||
|
|
"""
|
||
|
|
ref = sanitize_structured_reference(reference)
|
||
|
|
be_ref = re.match(r'(\d{10})(\d{2})', ref)
|
||
|
|
return be_ref and int(be_ref.group(1)) % 97 == int(be_ref.group(2)) % 97
|
||
|
|
|
||
|
|
def is_valid_structured_reference_fi(reference):
|
||
|
|
"""Check whether the provided reference is a valid structured reference for Finland.
|
||
|
|
|
||
|
|
:param reference: the reference to check
|
||
|
|
"""
|
||
|
|
ref = sanitize_structured_reference(reference)
|
||
|
|
fi_ref = re.match(r'(\d{1,19})(\d)', ref)
|
||
|
|
if not fi_ref:
|
||
|
|
return False
|
||
|
|
total = sum((7, 3, 1)[idx % 3] * int(val) for idx, val in enumerate(fi_ref.group(1)[::-1]))
|
||
|
|
check_digit = (10 - (total % 10)) % 10
|
||
|
|
return check_digit == int(fi_ref.group(2))
|
||
|
|
|
||
|
|
def is_valid_structured_reference_no_se(reference):
|
||
|
|
"""Check whether the provided reference is a valid structured reference for Norway or Sweden.
|
||
|
|
|
||
|
|
:param reference: the reference to check
|
||
|
|
"""
|
||
|
|
ref = sanitize_structured_reference(reference)
|
||
|
|
no_se_ref = re.match(r'\d+', ref)
|
||
|
|
return no_se_ref and luhn.is_valid(ref)
|