# Copyright (C) 2018 - TODAY, Pavlov Media
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo import api, fields, models


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

    agreement_ids = fields.One2many("agreement", "partner_id", string="Agreements")
    agreements_count = fields.Integer(compute="_compute_agreements_count")

    @api.depends("agreement_ids")
    def _compute_agreements_count(self):
        domain = [("partner_id", "in", self.ids)]
        res = self.env["agreement"].read_group(
            domain=domain, fields=["partner_id"], groupby=["partner_id"]
        )
        agreement_dict = {x["partner_id"][0]: x["partner_id_count"] for x in res}
        for rec in self:
            rec.agreements_count = agreement_dict.get(rec.id, 0)

    def action_open_agreement(self):
        self.ensure_one()
        action = self.env["ir.actions.actions"]._for_xml_id(
            "agreement.agreement_action"
        )
        action.update(
            {
                "domain": [("partner_id", "=", self.id)],
                "context": {"default_partner_id": self.id},
            }
        )
        return action