# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
import pytz

from odoo import api, models
from odoo.addons.resource.models.utils import Intervals


class HelpdeskTeam(models.Model):
    _inherit = 'helpdesk.team'

    @api.model
    def _get_working_user_interval(self, start_dt, end_dt, calendar, users, compute_leaves=True):
        leaves = {}
        if compute_leaves:
            leaves = self.env['hr.leave']._get_leave_interval(
                date_from=start_dt.astimezone(pytz.timezone('UTC')),
                date_to=end_dt.astimezone(pytz.timezone('UTC')),
                employee_ids=users.employee_id
            )
        # We do not pass compute_leaves as when True, need to take the non validated leaves into account,
        # which is done by calling _get_leave_interval on hr.leave (as it is not the case in _work_intervals_batch).
        res = super()._get_working_user_interval(start_dt, end_dt, calendar, users, False)
        for employee in users.employee_ids:
            intervals = res[employee.resource_id.id]
            employee_leaves = leaves.get(employee.id, [])
            for leave in employee_leaves:
                if intervals:
                    leave_intervals = Intervals([(
                        pytz.utc.localize(leave.date_from),
                        pytz.utc.localize(leave.date_to),
                        leave)]),
                    intervals -= leave_intervals[0]
            res[employee.resource_id.id] = intervals
        return res