# © 2020 Le Filament (<http://www.le-filament.com>)
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from odoo import fields, models


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

    ur_financial_system_id = fields.Many2one(
        comodel_name="ur.financial.system",
        string="Dispositif Financier",
        ondelete="set null",
    )
    ur_financial_system_date = fields.Date("Date de fin de dispositif")
    ur_regional_convention_id = fields.Many2one(
        comodel_name="ur.regional.convention",
        string="Convention Régionale",
        ondelete="set null",
    )
    ur_regional_convention_date = fields.Date("Date de fin de convention")
    ur_financial_system_nb = fields.Integer(
        string="Nb Dispositifs Financiers", compute="_compute_ur_system_nb"
    )
    ur_regional_convention_nb = fields.Integer(
        string="Nb conventions régionales", compute="_compute_ur_system_nb"
    )

    # ------------------------------------------------------
    # Compute Functions
    # ------------------------------------------------------
    def _compute_ur_system_nb(self):
        for partner in self:
            # Calcul nombre de dispositifs financiers
            financial_system = partner.env["ur.financial.system"].search(
                [("ur_id", "=", self.env.user.ur_id.id)]
            )
            partner.ur_financial_system_nb = len(financial_system)
            # Calcul nombre de conventions
            regional_convention = partner.env["ur.regional.convention"].search(
                [("ur_id", "=", self.env.user.ur_id.id)]
            )
            partner.ur_regional_convention_nb = len(regional_convention)