mirror of
https://github.com/OCA/contract.git
synced 2025-02-13 17:57:24 +02:00
83e7c0ee9c
[MIG] Prettier and Black [MIG] Prettier [MIG] Fixed Tracking [ADD]Added the Tests. [IMP]Improved the Tests.
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
# Copyright 2017-2020 Akretion France (http://www.akretion.com/)
|
|
# @author: Alexis de Lattre <alexis.delattre@akretion.com>
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
|
|
from odoo import fields, models
|
|
|
|
|
|
class Agreement(models.Model):
|
|
_inherit = "agreement"
|
|
|
|
invoice_ids = fields.One2many(
|
|
"account.move", "agreement_id", string="Invoices", readonly=True
|
|
)
|
|
out_invoice_count = fields.Integer(
|
|
compute="_compute_invoice_count", string="# of Customer Invoices"
|
|
)
|
|
in_invoice_count = fields.Integer(
|
|
compute="_compute_invoice_count", string="# of Vendor Bills"
|
|
)
|
|
|
|
def _compute_invoice_count(self):
|
|
base_domain = [("agreement_id", "in", self.ids)]
|
|
aio = self.env["account.move"]
|
|
out_rg_res = aio.read_group(
|
|
base_domain + [("move_type", "in", ("out_invoice", "out_refund"))],
|
|
["agreement_id"],
|
|
["agreement_id"],
|
|
)
|
|
out_data = {x["agreement_id"][0]: x["agreement_id_count"] for x in out_rg_res}
|
|
in_rg_res = aio.read_group(
|
|
base_domain + [("move_type", "in", ("in_invoice", "in_refund"))],
|
|
["agreement_id"],
|
|
["agreement_id"],
|
|
)
|
|
in_data = {x["agreement_id"][0]: x["agreement_id_count"] for x in in_rg_res}
|
|
for agreement in self:
|
|
agreement.out_invoice_count = out_data.get(agreement.id, 0)
|
|
agreement.in_invoice_count = in_data.get(agreement.id, 0)
|