83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
# © 2011-2019 Camptocamp SA
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html)
|
|
from odoo import _, fields, models
|
|
from odoo.addons.account_move_base_import.models.account_move import \
|
|
ErrorTooManyPartner
|
|
|
|
|
|
class AccountMoveCompletionRule(models.Model):
|
|
"""Add a rule based on transaction ID"""
|
|
|
|
_inherit = "account.move.completion.rule"
|
|
|
|
function_to_call = fields.Selection(
|
|
selection_add=[
|
|
('get_from_transaction_id_and_so',
|
|
'Match Sales Order using transaction ID'),
|
|
('get_from_transaction_id_and_invoice',
|
|
'Match Invoice using transaction ID')
|
|
])
|
|
|
|
def get_from_transaction_id_and_so(self, line):
|
|
"""
|
|
Match the partner based on the transaction ID field of the SO.
|
|
Then, call the generic st_line method to complete other values.
|
|
In that case, we always fullfill the reference of the line with the SO
|
|
name.
|
|
:param dict st_line: read of the concerned account.bank.statement.line
|
|
:return:
|
|
A dict of value that can be passed directly to the write method of
|
|
the statement line or {}
|
|
{'partner_id': value,
|
|
'account_id' : value,
|
|
...}
|
|
"""
|
|
res = {}
|
|
so_obj = self.env['sale.order']
|
|
sales = so_obj.search([('transaction_id', '=', line.ref)])
|
|
partners = sales.mapped('partner_id')
|
|
if len(partners) > 1:
|
|
raise ErrorTooManyPartner(
|
|
_('Line named "%s" was matched by more than '
|
|
'one partner.') % line.name)
|
|
if len(partners) == 1:
|
|
res['partner_id'] = partners.id
|
|
res['account_id'] = partners.property_account_receivable_id.id
|
|
return res
|
|
|
|
def get_from_transaction_id_and_invoice(self, line):
|
|
"""Match the partner based on the transaction ID field of the invoice.
|
|
Then, call the generic st_line method to complete other values.
|
|
|
|
In that case, we always fullfill the reference of the line with the
|
|
invoice name.
|
|
|
|
:param dict st_line: read of the concerned account.bank.statement.line
|
|
:return:
|
|
A dict of value that can be passed directly to the write method of
|
|
the statement line or {}
|
|
{'partner_id': value,
|
|
'account_id' : value,
|
|
...}
|
|
"""
|
|
res = {}
|
|
invoice_obj = self.env['account.invoice']
|
|
invoices = invoice_obj.search(
|
|
[('transaction_id', '=', line.ref)])
|
|
partners = invoices.mapped('partner_id.commercial_partner_id')
|
|
accounts = invoices.mapped('account_id')
|
|
if len(partners) > 1:
|
|
raise ErrorTooManyPartner(
|
|
_('Line named "%s" was matched by more than '
|
|
'one partner.') % line.name)
|
|
elif len(partners) == 1:
|
|
res['partner_id'] = partners.id
|
|
if len(accounts) > 1:
|
|
raise ErrorTooManyPartner(
|
|
_('Line named "%s" was matched by more than '
|
|
'one invoice.') % line.name
|
|
)
|
|
elif len(accounts) == 1:
|
|
res['account_id'] = accounts.id
|
|
return res
|