# Copyright 2015-2017 Odoo S.A.
# Copyright 2017 Tecnativa - Vicent Cubells
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).

from odoo import api, fields, models


class ResPartner(models.Model):
    _inherit = "res.partner"

    mgmtsystem_claim_count = fields.Integer(
        string="# Mgmt Claims", compute="_compute_mgmtsystem_claim_count"
    )
    mgmtsystem_claim_ids = fields.One2many(
        comodel_name="mgmtsystem.claim", inverse_name="partner_id"
    )

    @api.depends("claim_ids", "child_ids", "child_ids.claim_ids")
    def _compute_mgmtsystem_claim_count(self):
        partners = self | self.mapped("child_ids")
        partner_data = self.env["mgmtsystem.claim"].read_group(
            [("partner_id", "in", partners.ids)], ["partner_id"], ["partner_id"]
        )
        mapped_data = {m["partner_id"][0]: m["partner_id_count"] for m in partner_data}
        for partner in self:
            partner.mgmtsystem_claim_count = mapped_data.get(partner.id, 0)
            for child in partner.child_ids:
                partner.mgmtsystem_claim_count += mapped_data.get(child.id, 0)