odoo/addons/hr_contract/models/base_action_rule.py

28 lines
1.4 KiB
Python

# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import api, fields, models
from odoo.fields import Datetime
class BaseActionRule(models.Model):
""" Add resource and calendar for time-based conditions """
_inherit = 'base.action.rule'
trg_date_resource_field_id = fields.Many2one('ir.model.fields', string='Use employee work schedule', help='Use the user\'s working schedule.')
@api.model
def _check_delay(self, action, record, record_dt):
""" Override the check of delay to try to use a user-related calendar.
If no calendar is found, fallback on the default behavior.
"""
if action.trg_date_calendar_id and action.trg_date_range_type == 'day' and action.trg_date_resource_field_id:
user = record[action.trg_date_resource_field_id.name]
if user.employee_ids and user.employee_ids[0].contract_id and user.employee_ids[0].contract_id.working_hours:
calendar = user.employee_ids[0].contract_id.working_hours
start_dt = Datetime.from_string(record_dt)
resource_id = user.employee_ids[0].resource_id.id
return calendar.schedule_days_get_date(action.trg_date_range, day_date=start_dt, compute_leaves=True, resource_id=resource_id)
return super(BaseActionRule, self)._check_delay(action, record, record_dt)