You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
2.4 KiB
56 lines
2.4 KiB
# -*- coding: utf-8 -*-
|
|
###############################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2024-TODAY Cybrosys Technologies(<https://www.cybrosys.com>)
|
|
# Author: Gayathri V (odoo@cybrosys.com)
|
|
#
|
|
# You can modify it under the terms of the GNU AFFERO
|
|
# GENERAL PUBLIC LICENSE (AGPL v3), Version 3.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU AFFERO GENERAL PUBLIC LICENSE (AGPL v3) for more details.
|
|
#
|
|
# You should have received a copy of the GNU AFFERO GENERAL PUBLIC LICENSE
|
|
# (AGPL v3) along with this program.
|
|
# If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
###############################################################################
|
|
from odoo import models
|
|
|
|
|
|
class HrEmployee(models.Model):
|
|
"""Inherit the model hr.employee to add the functionality for retrieving
|
|
public holiday information"""
|
|
_inherit = 'hr.employee'
|
|
|
|
def get_public_holidays(self, start_date, end_date):
|
|
"""The function get_public_holidays takes in a start date and end
|
|
date as arguments and returns a dictionary with all the public
|
|
holidays within that range. It does this by calling the
|
|
_get_public_holidays method and then iterating through the results to
|
|
add each holiday to the dictionary."""
|
|
all_days = {}
|
|
user = self or self.env.user.employee_id
|
|
public_holidays = user._get_public_holidays(start_date, end_date)
|
|
for holiday in public_holidays:
|
|
num_days = (holiday.date_to - holiday.date_from).days
|
|
for day in range(num_days + 1):
|
|
all_days[str(holiday.date_from.date())] = day
|
|
return all_days
|
|
|
|
def _get_public_holidays(self, start_date, end_date):
|
|
"""The _get_public_holidays function searches for public holidays
|
|
within a given date range, for all companies associated with the
|
|
current environment's user. It returns a recordset of
|
|
resource.calendar.leaves that match the search criteria."""
|
|
public_holidays = self.env['resource.calendar.leaves'].search([
|
|
('date_from', '<=', end_date),
|
|
('date_to', '>=', start_date),
|
|
('resource_id', '=', False),
|
|
('company_id', 'in', self.env.companies.ids),
|
|
])
|
|
return public_holidays
|
|
|