Cédric Krier
3a1bdb65bf
Each country/bank has its own requirement about the ID to use on the initiator party so we add an option on the journal to define which identifier to use. issue8063 review284021004
123 lines
4.5 KiB
Python
123 lines
4.5 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
from sql import Null, Literal
|
|
from sql.functions import CurrentTimestamp
|
|
|
|
import stdnum.eu.at_02 as sepa
|
|
import stdnum.exceptions
|
|
|
|
from trytond.i18n import gettext
|
|
from trytond.pool import PoolMeta, Pool
|
|
from trytond.model import fields
|
|
from trytond.transaction import Transaction
|
|
from trytond.pyson import Eval
|
|
|
|
from trytond.modules.party.exceptions import InvalidIdentifierCode
|
|
from .exceptions import PartyIdentificationdError
|
|
|
|
|
|
class Party(metaclass=PoolMeta):
|
|
__name__ = 'party.party'
|
|
sepa_creditor_identifier_used = fields.Function(fields.Char(
|
|
'SEPA Creditor Identifier Used'),
|
|
'get_sepa_creditor_identifier_used')
|
|
sepa_mandates = fields.One2Many('account.payment.sepa.mandate', 'party',
|
|
'SEPA Mandates')
|
|
|
|
@classmethod
|
|
def __register__(cls, module_name):
|
|
Identifier = Pool().get('party.identifier')
|
|
cursor = Transaction().connection.cursor()
|
|
sql_table = cls.__table__()
|
|
identifier = Identifier.__table__()
|
|
super(Party, cls).__register__(module_name)
|
|
table = cls.__table_handler__(module_name)
|
|
|
|
# Migration from 4.0: Move sepa_creditor_identifier to identifier
|
|
if table.column_exist('sepa_creditor_identifier'):
|
|
select = sql_table.select(Literal(0), CurrentTimestamp(),
|
|
sql_table.id, Literal('sepa'),
|
|
sql_table.sepa_creditor_identifier,
|
|
where=((sql_table.sepa_creditor_identifier != Null)
|
|
& (sql_table.sepa_creditor_identifier != "")))
|
|
cursor.execute(*identifier.insert(
|
|
columns=[identifier.create_uid, identifier.create_date,
|
|
identifier.party, identifier.type, identifier.code],
|
|
values=select))
|
|
table.drop_column('sepa_creditor_identifier')
|
|
|
|
def get_sepa_creditor_identifier_used(self, name):
|
|
for identifier in self.identifiers:
|
|
if identifier.type == 'sepa':
|
|
return identifier.code
|
|
|
|
def get_sepa_identifier(self, name):
|
|
pool = Pool()
|
|
Identifier = pool.get('party.identifier')
|
|
for identifier in self.identifiers:
|
|
if identifier.type == name:
|
|
return identifier.sepa_identifier
|
|
else:
|
|
selection = Identifier.fields_get(['type'])['type']['selection']
|
|
type = dict(selection).get(name, name)
|
|
raise PartyIdentificationdError(
|
|
gettext('account_payment_sepa.msg_party_no_id',
|
|
party=self.rec_name,
|
|
type=type))
|
|
|
|
|
|
class PartyIdentifier(metaclass=PoolMeta):
|
|
__name__ = 'party.identifier'
|
|
|
|
sepa_es_suffix = fields.Char(
|
|
"SEPA Suffix", size=3,
|
|
states={
|
|
'invisible': Eval('type') != 'es_nif',
|
|
},
|
|
depends=['type'])
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(PartyIdentifier, cls).__setup__()
|
|
cls.type.selection.append(('sepa', 'SEPA Creditor Identifier'))
|
|
|
|
@fields.depends('party', '_parent_party.identifiers')
|
|
def check_code(self):
|
|
super(PartyIdentifier, self).check_code()
|
|
if self.type == 'sepa':
|
|
if not sepa.is_valid(self.code):
|
|
# Called from pre_validate so party may not be saved yet
|
|
if self.party and self.party.id > 0:
|
|
party = self.party.rec_name
|
|
else:
|
|
party = ''
|
|
raise InvalidIdentifierCode(
|
|
gettext('account_payment_sepa.msg_party_invalid_sepa',
|
|
code=self.code,
|
|
party=party))
|
|
|
|
@fields.depends('type', 'code')
|
|
def on_change_with_code(self):
|
|
code = super(PartyIdentifier, self).on_change_with_code()
|
|
if self.type == 'sepa':
|
|
try:
|
|
return sepa.compact(self.code)
|
|
except stdnum.exceptions.ValidationError:
|
|
pass
|
|
return code
|
|
|
|
@property
|
|
def sepa_identifier(self):
|
|
identifier = {
|
|
'Type': 'OrgId',
|
|
'Id': self.code,
|
|
}
|
|
if self.type == 'sepa':
|
|
identifier['Type'] = 'PrvtId'
|
|
identifier['SchmeNm'] = {'Prtry': 'SEPA'}
|
|
elif self.type == 'be_vat':
|
|
identifier['Issr'] = 'KBO-BCE'
|
|
elif self.type == 'es_nif':
|
|
identifier['Id'] += self.sepa_es_suffix or '000'
|
|
return identifier
|