contract/contract_invoice_merge_by_partner/models/account_analytic_analysis.py

# -*- coding: utf-8 -*-
# Copyright 2016 Carlos Dauden <carlos.dauden@tecnativa.com>
# Copyright 2016 Pedro M. Baeza <pedro.baeza@tecnativa.com>
# Copyright 2017 Vicent Cubells <vicent.cubells@tecnativa.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from openerp import api, models


class AccountAnalyticAccount(models.Model):
    _inherit = 'account.analytic.account'

    @api.multi
    def recurring_create_invoice(self):
        invoices = super(
            AccountAnalyticAccount, self).recurring_create_invoice()
        invoices_info = {}
        invoices2unlink = AccountInvoice = self.env['account.invoice']
        for partner in invoices.mapped('partner_id'):
            invoices2merge = invoices.filtered(
                lambda x: x.partner_id == partner)
            if partner.contract_invoice_merge and len(invoices2merge) > 1:
                invoices_info.update(invoices2merge.do_merge())
                invoices2unlink += invoices2merge
        invoices -= invoices2unlink
        invoices2unlink.unlink()
        return invoices | AccountInvoice.browse(invoices_info.keys())