62 lines
1.9 KiB
Python
Executable File
62 lines
1.9 KiB
Python
Executable File
#This file is part party_bank_validation module for Tryton.
|
|
#The COPYRIGHT file at the top level of this repository contains
|
|
#the full copyright notices and license terms.
|
|
import logging
|
|
from trytond.model import fields
|
|
from trytond.pool import PoolMeta
|
|
|
|
HAS_BANKNUMBER = False
|
|
BANK_COUNTRIES = [('', '')]
|
|
try:
|
|
import banknumber
|
|
HAS_BANKNUMBER = True
|
|
for country in banknumber.countries():
|
|
BANK_COUNTRIES.append((country, country))
|
|
except ImportError:
|
|
logging.getLogger('party_bank_validation').warning(
|
|
'Unable to import banknumber. Bank code validation disabled.')
|
|
|
|
|
|
__all__ = ['BankAccount']
|
|
__metaclass__ = PoolMeta
|
|
|
|
|
|
class BankAccount:
|
|
'Bank Account'
|
|
__name__ = 'bank.account'
|
|
bank_country = fields.Selection(BANK_COUNTRIES, 'Bank Country',
|
|
help="Setting Bank country will enable validation of the Bank code.")
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(BankAccount, cls).__setup__()
|
|
cls._error_messages.update({
|
|
'invalid_bank_number': 'Invalid bank number in account "%s".',
|
|
})
|
|
|
|
@classmethod
|
|
def validate(cls, accounts):
|
|
for account in accounts:
|
|
account.check_bank_number()
|
|
|
|
def check_bank_number(self):
|
|
'''
|
|
Check the Bank number depending of the country.
|
|
'''
|
|
if not HAS_BANKNUMBER:
|
|
return
|
|
if not self.bank_country:
|
|
return
|
|
|
|
code = self.code.replace(' ','')
|
|
if (not getattr(banknumber, 'check_code_'
|
|
+ self.bank_country.lower())(code)):
|
|
#Check if user doesn't have put country code in code
|
|
if code.startswith(self.bank_country):
|
|
code = code[len(self.bank_country):]
|
|
BankAccount.write([self], {
|
|
'code': code,
|
|
})
|
|
else:
|
|
self.raise_user_error('invalid_bank_number', (self.rec_name,))
|