mirror of
https://gitlab.com/flectra-community/bank-payment.git
synced 2024-11-22 13:42:07 +00:00
51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
# Copyright 2016 Akretion (Alexis de Lattre <alexis.delattre@akretion.com>)
|
|
# Copyright 2017 Carlos Dauden <carlos.dauden@tecnativa.com>
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
|
|
|
|
from flectra import models, fields, api
|
|
|
|
|
|
class ResPartner(models.Model):
|
|
_inherit = 'res.partner'
|
|
|
|
mandate_count = fields.Integer(
|
|
compute='_compute_mandate_count', string="Number of Mandates",
|
|
readonly=True)
|
|
valid_mandate_id = fields.Many2one(
|
|
comodel_name='account.banking.mandate',
|
|
compute='_compute_valid_mandate_id',
|
|
string='First Valid Mandate')
|
|
|
|
@api.multi
|
|
def _compute_mandate_count(self):
|
|
mandate_data = self.env['account.banking.mandate'].read_group(
|
|
[('partner_id', 'in', self.ids)], ['partner_id'], ['partner_id'])
|
|
mapped_data = dict([
|
|
(mandate['partner_id'][0], mandate['partner_id_count'])
|
|
for mandate in mandate_data])
|
|
for partner in self:
|
|
partner.mandate_count = mapped_data.get(partner.id, 0)
|
|
|
|
@api.multi
|
|
def _compute_valid_mandate_id(self):
|
|
# Dict for reducing the duplicated searches on parent/child partners
|
|
company_id = self.env.context.get('force_company', False)
|
|
if company_id:
|
|
company = self.env['res.company'].browse(company_id)
|
|
else:
|
|
company = self.env['res.company']._company_default_get(
|
|
'account.banking.mandate')
|
|
|
|
mandates_dic = {}
|
|
for partner in self:
|
|
commercial_partner_id = partner.commercial_partner_id.id
|
|
if commercial_partner_id in mandates_dic:
|
|
partner.valid_mandate_id = mandates_dic[commercial_partner_id]
|
|
else:
|
|
mandates = partner.commercial_partner_id.bank_ids.mapped(
|
|
'mandate_ids').filtered(
|
|
lambda x: x.state == 'valid' and x.company_id == company)
|
|
first_valid_mandate_id = mandates[:1].id
|
|
partner.valid_mandate_id = first_valid_mandate_id
|
|
mandates_dic[commercial_partner_id] = first_valid_mandate_id
|