Medium Article - HR Leave - Model Override

hr.leave.py

from collections import defaultdict
from odoo import models, api


class HrLeave(models.Model):
    _inherit = "hr.leave"
    
    @api.model
    def get_employee_timeoff_periods(self, res_model, res_ids):
        if res_model == "planning.slot":
            # the res_ids supplied is rather the resource ids
            # and not the employee ids, let therefore infer the
            # employee ids from the reosurce records
            employees = self.env["hr.employee"].search([("resource_id", "in", res_ids)])
            employee_ids = employees.ids
        else:
            employee_ids = res_ids
        timeoffs = self.search([("employee_id", "in", employee_ids)])
        timeoff_periods_per_employee = defaultdict(list)
        for t in timeoffs:
            if res_model == "planning.slot":
                res_id = t.employee_id.resource_id.id
            else:
                res_id = t.employee_id.id
            timeoff_periods_per_employee[res_id].append(
                [
                    t.request_date_from,
                    t.request_date_to,
                    t.holiday_status_id.name,
                    t.holiday_status_id.icon_id.name,
                ]
            )
        return timeoff_periods_per_employee