odoo/ext/custom-addons/hr_holidays_compute_days/models/hr_employee.py

21 lines
763 B
Python

# Copyright 2017-2018 Tecnativa - Pedro M. Baeza
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from odoo import models
class HrEmployee(models.Model):
_inherit = 'hr.employee'
def get_day_work_hours_count(self, day_date, calendar=None):
"""Return 24 as work hours when detecting through context and weekday
that the day passed is one of the rest days.
"""
if self.env.context.get('include_rest_days') and calendar:
real_weekdays = calendar.with_context(
include_rest_days=False,
)._get_weekdays()
if day_date.weekday() not in real_weekdays:
return 24
return super().get_day_work_hours_count(day_date, calendar=calendar)