import logging
import pytz
from collections import defaultdict
from datetime import datetime, timedelta, time
from dateutil.relativedelta import relativedelta
from math import ceil
from markupsafe import Markup
from odoo.addons.base.models.ir_model import MODULE_UNINSTALL_FLAG
from odoo.addons.resource.models.utils import HOURS_PER_DAY
from odoo import api, fields, models
from odoo.addons.base.models.res_partner import _tz_get
from odoo.exceptions import AccessError, UserError, ValidationError
from odoo.tools.date_utils import float_to_time
from odoo.fields import Command, Date, Domain
from odoo.tools.float_utils import float_round, float_compare
from odoo.tools.intervals import Intervals
from odoo.tools.misc import clean_context, format_date
from odoo.tools.translate import _
_logger = logging.getLogger(__name__)
def get_employee_from_context(values, context, user_employee_id):
employee_ids_list = [value[2] for value in values.get('employee_ids', []) if len(value) == 3 and value[0] == Command.SET]
employee_ids = employee_ids_list[-1] if employee_ids_list else []
employee_id_value = employee_ids[0] if employee_ids else False
return employee_id_value or context.get('default_employee_id', context.get('employee_id', user_employee_id))
class HrLeave(models.Model):
""" Time Off Requests Access specifications
- a regular employee / user
- can see all leaves;
- cannot see name field of leaves belonging to other user as it may contain
private information that we don't want to share to other people than
HR people;
- can modify only its own not validated leaves (except writing on state to
bypass approval);
- can discuss on its leave requests;
- can reset only its own leaves;
- cannot validate any leaves;
- an Officer
- can see all leaves;
- can validate "HR" single validation leaves from people if
- he is the employee manager;
- he is the department manager;
- he is member of the same department;
- target employee has no manager and no department manager;
- can validate "Manager" single validation leaves from people if
- he is the employee manager;
- he is the department manager;
- target employee has no manager and no department manager;
- can first validate "Both" double validation leaves from people like "HR"
single validation, moving the leaves to validate1 state;
- cannot validate its own leaves;
- can reset only its own leaves;
- can refuse all leaves;
- a Manager
- can do everything he wants
On top of that multicompany rules apply based on company defined on the
leave request leave type.
"""
_name = 'hr.leave'
_description = "Time Off"
_order = "date_from desc"
_inherit = ['mail.thread.main.attachment', 'mail.activity.mixin']
_mail_post_access = 'read'
@api.model
def default_get(self, fields):
defaults = super().default_get(fields)
defaults = self._default_get_request_dates(defaults)
if self.env.context.get('holiday_status_display_name', True) and 'holiday_status_id' in fields and not defaults.get('holiday_status_id'):
domain = ['|', ('requires_allocation', '=', False), ('has_valid_allocation', '=', True)]
defaults['holiday_status_id'] = False
leave_types = self.env['hr.leave.type'].search(domain, order='sequence')
selected_leave_type = next(
(
leave_type for leave_type in leave_types
if (defaults.get('request_unit_hours') and leave_type['request_unit'] == 'hour') or (not defaults.get('request_unit_hours'))
),
leave_types[0] if leave_types else None,
)
if selected_leave_type:
defaults['holiday_status_id'] = selected_leave_type.id
defaults['request_unit_hours'] = (selected_leave_type.request_unit == 'hour')
if 'request_date_from' in fields and 'request_date_from' not in defaults:
defaults['request_date_from'] = Date.today()
if 'request_date_to' in fields and 'request_date_to' not in defaults:
defaults['request_date_to'] = Date.today()
return defaults
def _default_get_request_dates(self, values):
# The UI views initialize date_{from,to} due to how calendar views work.
# However it is request_date_{from,to} that should be used instead.
# Instead of overwriting all the javascript methods to use
# request_date_{from,to} instead of date_{from,to}, we just convert
# date_{from,to} to request_date_{from,to} here.
# Request dates are determined during an onchange scenario.
# To ensure that the values are correct in the client context (UI),
# the timezone must be applied (because no processing is carried out
# when these dates are received on the frontend).
# Note:
# Without the application of the timezone, days based on UTC datetimes
# will be returned (and will therefore not be correct for the client).
client_tz = self.env.tz
if values.get('date_from'):
if not values.get('request_date_from'):
values['request_date_from'] = pytz.utc.localize(values['date_from']).astimezone(client_tz)
del values['date_from']
if values.get('date_to'):
if not values.get('request_date_to'):
values['request_date_to'] = pytz.utc.localize(values['date_to']).astimezone(client_tz)
del values['date_to']
return values
# description
name = fields.Char('Description', compute='_compute_description', inverse='_inverse_description', search='_search_description', compute_sudo=False, copy=False)
private_name = fields.Char('Time Off Description', groups='hr_holidays.group_hr_holidays_responsible')
state = fields.Selection([
('confirm', 'To Approve'),
('refuse', 'Refused'),
('validate1', 'Second Approval'),
('validate', 'Approved'),
('cancel', 'Cancelled'),
], string='Status', store=True, tracking=True, copy=False, readonly=False, default='confirm')
user_id = fields.Many2one('res.users', string='User', related='employee_id.user_id', related_sudo=True, compute_sudo=True, store=True, readonly=True, index=True)
# leave type configuration
holiday_status_id = fields.Many2one(
"hr.leave.type", compute='_compute_from_employee_id',
store=True, string="Time Off Type",
required=True, readonly=False,
domain="""[
'|',
('requires_allocation', '=', False),
('has_valid_allocation', '=', True),
]""",
tracking=True)
holiday_status_requires_allocation = fields.Boolean(related="holiday_status_id.requires_allocation")
color = fields.Integer("Color", related='holiday_status_id.color')
validation_type = fields.Selection(string='Validation Type', related='holiday_status_id.leave_validation_type', readonly=False)
# HR data
employee_id = fields.Many2one(
'hr.employee', string='Employee', index=True, ondelete="restrict", required=True,
tracking=True, domain=lambda self: self._get_employee_domain(), default=lambda self: self.env.user.employee_id)
employee_company_id = fields.Many2one(related='employee_id.company_id', string="Employee Company", store=True)
company_id = fields.Many2one('res.company', compute='_compute_company_id', store=True)
active_employee = fields.Boolean(related='employee_id.active', string='Employee Active')
tz_mismatch = fields.Boolean(compute='_compute_tz_mismatch')
tz = fields.Selection(_tz_get, compute='_compute_tz')
department_id = fields.Many2one(
'hr.department', compute='_compute_department_id', store=True, string='Department',
readonly=False)
notes = fields.Text('Reasons', readonly=False)
# duration
resource_calendar_id = fields.Many2one('resource.calendar', compute='_compute_resource_calendar_id', store=True, readonly=False, copy=False)
# allocated leave balance
max_leaves = fields.Float(compute='_compute_leaves')
virtual_remaining_leaves = fields.Float(compute='_compute_leaves', string='Available Time Off')
# These dates are computed based on request_date_{to,from} and should
# therefore never be set directly.
date_from = fields.Datetime(
'Start Date', compute='_compute_date_from_to', store=True, index=True, tracking=True)
date_to = fields.Datetime(
'End Date', compute='_compute_date_from_to', store=True, tracking=True)
number_of_days = fields.Float(
'Duration (Days)', compute='_compute_duration', store=True, tracking=True,
help='Number of days of the time off request. Used in the calculation.')
number_of_hours = fields.Float(
'Duration (Hours)', compute='_compute_duration', store=True, tracking=True,
help='Number of hours of the time off request. Used in the calculation.')
last_several_days = fields.Boolean("All day", compute="_compute_last_several_days")
duration_display = fields.Char('Requested', compute='_compute_duration_display', store=True) # details
# details
meeting_id = fields.Many2one('calendar.event', string='Meeting', copy=False)
first_approver_id = fields.Many2one(
'hr.employee', string='First Approval', readonly=True, copy=False,
help='This area is automatically filled by the user who validate the time off')
second_approver_id = fields.Many2one(
'hr.employee', string='Second Approval', readonly=True, copy=False,
help='This area is automatically filled by the user who validate the time off with second level (If time off type need second validation)')
can_approve = fields.Boolean(compute='_compute_can_approve', export_string_translation=False)
can_validate = fields.Boolean(compute='_compute_can_validate', export_string_translation=False)
can_refuse = fields.Boolean(compute='_compute_can_refuse', export_string_translation=False)
can_cancel = fields.Boolean(compute='_compute_can_cancel', export_string_translation=False)
can_back_to_approve = fields.Boolean(compute='_compute_can_back_to_approve', export_string_translation=False)
attachment_ids = fields.One2many('ir.attachment', 'res_id', string="Attachments")
# To display in form view
supported_attachment_ids = fields.Many2many(
'ir.attachment', string="Attach File", compute='_compute_supported_attachment_ids',
inverse='_inverse_supported_attachment_ids')
supported_attachment_ids_count = fields.Integer(compute='_compute_supported_attachment_ids')
# UX fields
leave_type_request_unit = fields.Selection(related='holiday_status_id.request_unit', readonly=True)
leave_type_support_document = fields.Boolean(related="holiday_status_id.support_document")
# Interface fields used when not using hour-based computation
# These are the fields that should be used to manipulate the start- and
# end-dates of the leave request. date_from and date_to are computed and
# should therefore not be set directly.
request_date_from = fields.Date('Request Start Date')
request_date_to = fields.Date('Request End Date')
# Interface fields used when using hour-based computation
request_hour_from = fields.Float(string='Hour from', compute='_compute_request_hour_from_to', readonly=False, store=True)
request_hour_to = fields.Float(string='Hour to', compute='_compute_request_hour_from_to', readonly=False, store=True)
# used only when the leave is taken in half days
request_date_from_period = fields.Selection([
('am', 'Morning'), ('pm', 'Afternoon')],
string="Date Period Start", default='am')
request_date_to_period = fields.Selection([
('am', 'Morning'), ('pm', 'Afternoon')],
string="Date Period End", default='pm')
# request type
request_unit_half = fields.Boolean('Half-Day', compute='_compute_request_unit_half', store=True)
request_unit_hours = fields.Boolean('Specific Time', compute='_compute_request_unit_hours', store=True)
# view
is_hatched = fields.Boolean('Hatched', compute='_compute_is_hatched')
is_striked = fields.Boolean('Striked', compute='_compute_is_hatched')
has_mandatory_day = fields.Boolean(compute='_compute_has_mandatory_day')
leave_type_increases_duration = fields.Char(compute='_compute_leave_type_increases_duration')
# warning message
dashboard_warning_message = fields.Char(compute='_compute_dashboard_warning_message')
_date_check2 = models.Constraint(
'CHECK ((date_from <= date_to))',
'The start date must be before or equal to the end date.',
)
_date_check3 = models.Constraint(
'CHECK ((request_date_from <= request_date_to))',
'The request start date must be before or equal to the request end date.',
)
_duration_check = models.Constraint(
'CHECK ( number_of_days >= 0 )',
"If you want to change the number of days you should use the 'period' mode",
)
_date_to_date_from_index = models.Index("(date_to, date_from)")
@api.onchange('request_hour_from', 'request_hour_to')
def _onchange_hours(self):
# avoid negative or after midnight
self.request_hour_from = min(max(self.request_hour_from, 0.0), 23.99)
self.request_hour_to = min(max(self.request_hour_to, 0.0), 24)
@api.depends('employee_id', 'request_date_from', 'request_date_to', 'request_unit_hours')
def _compute_request_hour_from_to(self):
env_company_calendar = self.env.company.resource_calendar_id
for leave in self:
calendar = leave.resource_calendar_id or env_company_calendar
if (not leave.request_unit_hours
and leave.employee_id
and leave.request_date_from
and leave.request_date_to
and calendar):
hour_from, hour_to = leave._get_hour_from_to(leave.request_date_from, leave.request_date_to)
leave.request_hour_from = hour_from
leave.request_hour_to = hour_to
@api.depends('employee_id', 'leave_type_request_unit', 'request_date_from', 'request_date_to',
'request_hour_from', 'request_hour_to', 'request_date_from_period', 'request_date_to_period')
def _compute_dashboard_warning_message(self):
all_leaves = self.search([
('date_from', '<', max(self.mapped('date_to'))),
('date_to', '>', min(self.mapped('date_from'))),
('employee_id', 'in', self.employee_id.ids),
('holiday_status_id.allow_request_on_top', '=', False),
('state', 'not in', ['cancel', 'refuse']),
])
self.filtered(lambda self: self.state in ['cancel', 'refuse']).dashboard_warning_message = False
for holiday in self.filtered(lambda self: self.state not in ['cancel', 'refuse']):
conflicting_holidays = all_leaves.filtered_domain([
('employee_id', 'in', holiday.employee_id.ids),
('date_from', '<', holiday.date_to),
('date_to', '>', holiday.date_from),
('id', 'not in', holiday.ids),
])
if not conflicting_holidays:
holiday.dashboard_warning_message = False
continue
conflicting_holidays_list = []
# Do not display the name of the employee if the conflicting holidays have an employee_id.user_id equivalent to the user id
holidays_only_have_uid = bool(holiday.employee_id)
holiday_states = dict(conflicting_holidays.fields_get(allfields=['state'])['state']['selection'])
for conflicting_holiday in conflicting_holidays:
conflicting_holiday_data = {
'employee_name': conflicting_holiday.employee_id.name,
'date_from': format_date(self.env, min(conflicting_holiday.mapped('date_from'))),
'date_to': format_date(self.env, min(conflicting_holiday.mapped('date_to'))),
'state': holiday_states[conflicting_holiday.state]
}
if conflicting_holiday.employee_id.user_id.id != self.env.uid:
holidays_only_have_uid = False
if conflicting_holiday_data not in conflicting_holidays_list:
conflicting_holidays_list.append(conflicting_holiday_data)
msg = ""
if holidays_only_have_uid:
msg = self.env._('You\'ve already booked time off which overlaps with this period:')
else:
msg = self.env._('An employee already booked time off which overlaps with this period:')
holiday.dashboard_warning_message = msg + "".join(
('\n\t' + self.env._('%(employee_name)s from %(date_from)s to %(date_to)s - %(state)s')) % {
'employee_name': conflicting_holiday_data['employee_name'] if not holidays_only_have_uid else "",
'date_from': conflicting_holiday_data['date_from'],
'date_to': conflicting_holiday_data['date_to'],
'state': conflicting_holiday_data['state']
} for conflicting_holiday_data in conflicting_holidays_list
)
@api.depends_context('uid')
def _compute_description(self):
self.check_access('read')
is_officer = self.env.user.has_group('hr_holidays.group_hr_holidays_user')
for leave in self:
if is_officer or leave.user_id == self.env.user or leave.employee_id.leave_manager_id == self.env.user:
leave.name = leave.sudo().private_name
else:
leave.name = '*****'
def _inverse_description(self):
is_officer = self.env.user.has_group('hr_holidays.group_hr_holidays_user')
for leave in self:
if is_officer or leave.user_id == self.env.user or leave.employee_id.leave_manager_id == self.env.user:
leave.sudo().private_name = leave.name
def _search_description(self, operator, value):
is_officer = self.env.user.has_group('hr_holidays.group_hr_holidays_user')
domain = Domain('private_name', operator, value)
if not is_officer:
domain &= Domain('user_id', '=', self.env.user.id)
query = self.sudo()._search(domain)
return Domain('id', 'in', query)
@api.depends('employee_id', 'request_date_from', 'request_date_to')
def _compute_resource_calendar_id(self):
leaves_without_emp_or_date = self.filtered(
lambda leave: not (leave.employee_id and leave.request_date_from and leave.request_date_to)
)
valid_leaves = self - leaves_without_emp_or_date
leaves_without_emp_or_date.resource_calendar_id = self.env.company.resource_calendar_id
if not valid_leaves:
return
employees_by_dates = defaultdict(lambda: self.env['hr.employee'])
contracts_by_employee = dict(
self.env['hr.version']._read_group(
domain=[('employee_id', 'in', self.employee_id.ids)],
groupby=['employee_id'],
aggregates=['id:recordset']
)
)
for leave in valid_leaves:
employees_by_dates[leave.request_date_from] += leave.employee_id
calendar_by_dates = {date_from: employees._get_calendars(date_from) for date_from, employees in employees_by_dates.items()}
for leave in valid_leaves:
calendar = calendar_by_dates.get(leave.request_date_from, {}).get(leave.employee_id.id) \
or self.env.company.resource_calendar_id
# We use the request dates to find the contracts, because date_from
# and date_to are not set yet at this point. Since these dates are
# used to get the contracts for which these leaves apply and
# contract start- and end-dates are just dates (and not datetimes)
# these dates are comparable.
contracts = contracts_by_employee.get(leave.employee_id, self.env['hr.version']).filtered(
lambda c: c.date_start <= leave.request_date_to and
(not c.date_end or c.date_end >= leave.request_date_from))
if contracts:
# If there are more than one contract they should all have the
# same calendar, otherwise a constraint is violated.
calendar = contracts[:1].resource_calendar_id
leave.resource_calendar_id = calendar
def _get_overlapping_contracts(self):
self.ensure_one()
domain = Domain.AND([
Domain('employee_id', '=', self.employee_id.id),
Domain('contract_date_start', '<=', self.date_to),
Domain.OR([
Domain('contract_date_end', '>=', self.date_from),
Domain('contract_date_end', '=', False),
])
])
versions = self.env['hr.version'].sudo().search(domain)
return versions.filtered(lambda v: v._is_overlapping_period(self.date_from.date(), self.date_to.date()))
@api.constrains('date_from', 'date_to')
def _check_contracts(self):
"""
A leave cannot be set across multiple contracts.
Note: a leave can be across multiple contracts despite this constraint.
It happens if a leave is correctly created (not across multiple contracts) but
contracts are later modifed/created in the middle of the leave.
"""
for holiday in self.filtered('employee_id'):
versions = holiday._get_overlapping_contracts()
if len(versions.resource_calendar_id) > 1:
raise ValidationError(
self.env._("""A leave cannot be set across multiple versions with different working schedules.
Please create one time off for each version period.
Time off:
%(time_off)s
Versions:
%(versions)s""",
time_off=holiday.display_name,
versions='\n'.join(_(
"- '%(version)s' from %(start_date)s to %(end_date)s",
version=version.name or version.employee_id.name,
start_date=format_date(self.env, version.date_start),
end_date=format_date(self.env, version.date_end) if version.date_end else self.env._("undefined"),
) for version in versions)))
@api.depends('request_date_from_period', 'request_date_to_period', 'request_hour_from', 'request_hour_to',
'request_date_from', 'request_date_to', 'request_unit_half', 'request_unit_hours', 'employee_id')
def _compute_date_from_to(self):
for holiday in self:
if not holiday.request_date_from:
holiday.date_from = False
continue
if not holiday.request_date_to:
holiday.date_to = False
continue
if holiday.request_unit_hours:
hour_from = holiday.request_hour_from
hour_to = holiday.request_hour_to
if not hour_from or not hour_to:
computed_from, computed_to = holiday._get_hour_from_to(holiday.request_date_from, holiday.request_date_to)
hour_from = hour_from or computed_from
hour_to = hour_to or computed_to
elif holiday.request_unit_half:
period_map = {'am': 'morning', 'pm': 'afternoon'}
from_period = period_map.get(holiday.request_date_from_period)
to_period = period_map.get(holiday.request_date_to_period)
if holiday.request_date_from == holiday.request_date_to:
day_period = from_period if from_period == to_period else None
hour_from, hour_to = holiday._get_hour_from_to(holiday.request_date_from, holiday.request_date_to,
day_period)
else:
hour_from, _ = holiday._get_hour_from_to(holiday.request_date_from, holiday.request_date_from, from_period)
_, hour_to = holiday._get_hour_from_to(holiday.request_date_to, holiday.request_date_to, to_period)
else:
hour_from, hour_to = holiday._get_hour_from_to(holiday.request_date_from, holiday.request_date_to)
holiday.date_from = self._to_utc(holiday.request_date_from, hour_from, holiday.employee_id or holiday)
holiday.date_to = self._to_utc(holiday.request_date_to, hour_to, holiday.employee_id or holiday)
@api.depends('leave_type_request_unit')
def _compute_request_unit_half(self):
for holiday in self:
holiday.request_unit_half = holiday.leave_type_request_unit == 'half_day'
@api.depends('leave_type_request_unit')
def _compute_request_unit_hours(self):
for holiday in self:
holiday.request_unit_hours = holiday.leave_type_request_unit == 'hour'
def _get_employee_domain(self):
domain = [
('active', '=', True),
('company_id', 'in', self.env.companies.ids),
]
if not self.env.user.has_group('hr_holidays.group_hr_holidays_user'):
domain += [
'|',
('user_id', '=', self.env.uid),
('leave_manager_id', '=', self.env.uid),
]
return domain
@api.depends('employee_id')
def _compute_from_employee_id(self):
for holiday in self:
if not holiday.holiday_status_id.requires_allocation:
continue
if not holiday.employee_id:
holiday.holiday_status_id = False
elif holiday.employee_id.user_id != self.env.user and holiday._origin.employee_id != holiday.employee_id:
if holiday.employee_id and not holiday.holiday_status_id.with_context(employee_id=holiday.employee_id.id).has_valid_allocation:
holiday.holiday_status_id = False
@api.depends('employee_id')
def _compute_department_id(self):
for holiday in self:
holiday.department_id = holiday.employee_id.department_id
@api.depends('date_from', 'date_to', 'holiday_status_id')
def _compute_has_mandatory_day(self):
date_from, date_to = min(self.mapped('date_from')), max(self.mapped('date_to'))
if date_from and date_to:
# Sudo to get access to version fields on employee (job_id)
mandatory_days = self.employee_id.sudo()._get_mandatory_days(
date_from.date(),
date_to.date())
for leave in self:
department_ids = leave.employee_id.department_id.ids
domain = [
('start_date', '<=', leave.date_to.date()),
('end_date', '>=', leave.date_from.date()),
'|',
('resource_calendar_id', '=', False),
('resource_calendar_id', '=', leave.resource_calendar_id.id),
]
if department_ids:
domain += [
'|',
('department_ids', '=', False),
('department_ids', 'parent_of', department_ids),
]
else:
domain += [('department_ids', '=', False)]
if leave.holiday_status_id.company_id:
domain += [('company_id', '=', leave.holiday_status_id.company_id.id)]
leave.has_mandatory_day = leave.date_from and leave.date_to and mandatory_days.filtered_domain(domain)
else:
self.has_mandatory_day = False
@api.depends('leave_type_request_unit', 'number_of_days')
def _compute_leave_type_increases_duration(self):
durations = self._get_durations(check_leave_type=False)
for leave in self:
days = durations[leave.id][0]
if leave.leave_type_request_unit == 'day' and leave.holiday_status_requires_allocation and days < leave.number_of_days:
leave.leave_type_increases_duration = self.env._("According to your working schedule you are expected to work"
" %(days)s days in this period, but %(nb_days)s days will be used because this leave"
" %(leave_type_name)s can only be taken by days.",
days=days, nb_days=leave.number_of_days, leave_type_name=leave.holiday_status_id.name)
else:
leave.leave_type_increases_duration = ''
def _get_durations(self, check_leave_type=True, resource_calendar=None):
"""
This method is factored out into a separate method from
_compute_duration so it can be hooked and called without necessarily
modifying the fields and triggering more computes of fields that
depend on number_of_hours or number_of_days.
"""
result = {}
employee_leaves = self.filtered('employee_id')
employees_by_dates_calendar = defaultdict(lambda: self.env['hr.employee'])
for leave in employee_leaves:
if not leave.date_from or not leave.date_to:
continue
employees_by_dates_calendar[(leave.date_from, leave.date_to, leave.holiday_status_id.include_public_holidays_in_duration, resource_calendar or leave.resource_calendar_id)] += leave.employee_id
# We force the company in the domain as we are more than likely in a compute_sudo
domain = [('time_type', '=', 'leave'),
('company_id', 'in', self.env.companies.ids + self.env.context.get('allowed_company_ids', [])),
# When searching for resource leave intervals, we exclude the one that
# is related to the leave we're currently trying to compute for.
'|', ('holiday_id', '=', False), ('holiday_id', 'not in', employee_leaves.ids)]
# Precompute values in batch for performance purposes
work_time_per_day_mapped = {
(date_from, date_to, include_public_holidays_in_duration, calendar): employees.with_context(
compute_leaves=not include_public_holidays_in_duration)._list_work_time_per_day(date_from, date_to, domain=domain, calendar=calendar)
for (date_from, date_to, include_public_holidays_in_duration, calendar), employees in employees_by_dates_calendar.items()
}
work_days_data_mapped = {
(date_from, date_to, include_public_holidays_in_duration, calendar): employees._get_work_days_data_batch(date_from, date_to, compute_leaves=not include_public_holidays_in_duration, domain=domain, calendar=calendar)
for (date_from, date_to, include_public_holidays_in_duration, calendar), employees in employees_by_dates_calendar.items()
}
for leave in self:
calendar = resource_calendar or leave.resource_calendar_id
if not leave.date_from or not leave.date_to or (not calendar and not leave.employee_id):
result[leave.id] = (0, 0)
continue
hours, days = (0, 0)
if leave.employee_id:
# For flexible employees, if it's a single day leave, we force it to the real duration since the virtual intervals might not match reality on that day, especially for custom hours
# sudo as is_flexible is on version model and employee does not have access to it.
if leave.employee_id.sudo().is_flexible and leave.request_date_to == leave.request_date_from:
public_holidays = self.env['resource.calendar.leaves'].search([
('resource_id', '=', False),
('date_from', '<', leave.date_to),
('date_to', '>', leave.date_from),
('calendar_id', 'in', [False, calendar.id]),
('company_id', '=', leave.company_id.id)
])
if public_holidays:
public_holidays_intervals = Intervals([(ph.date_from, ph.date_to, ph) for ph in public_holidays])
leave_intervals = Intervals([(leave.date_from, leave.date_to, leave)])
real_leave_intervals = leave_intervals - public_holidays_intervals
hours = 0
for start, stop, meta in real_leave_intervals:
hours += (stop - start).total_seconds() / 3600
else:
hours = (leave.date_to - leave.date_from).total_seconds() / 3600
if not leave.request_unit_hours and not public_holidays:
days = 1 if not leave.request_unit_half or leave.request_date_from_period != leave.request_date_to_period else 0.5
else:
days = hours / 24
elif leave.leave_type_request_unit == 'day' and check_leave_type:
# list of tuples (day, hours)
work_time_per_day_list = work_time_per_day_mapped[leave.date_from, leave.date_to, leave.holiday_status_id.include_public_holidays_in_duration, calendar][leave.employee_id.id]
days = len(work_time_per_day_list)
hours = sum(map(lambda t: t[1], work_time_per_day_list))
else:
work_days_data = work_days_data_mapped[leave.date_from, leave.date_to, leave.holiday_status_id.include_public_holidays_in_duration, calendar][leave.employee_id.id]
hours, days = work_days_data['hours'], work_days_data['days']
else:
today_hours = calendar.get_work_hours_count(
datetime.combine(leave.date_from.date(), time.min),
datetime.combine(leave.date_from.date(), time.max),
False)
hours = calendar.get_work_hours_count(leave.date_from, leave.date_to, compute_leaves=not leave.holiday_status_id.include_public_holidays_in_duration)
days = hours / (today_hours or HOURS_PER_DAY)
if leave.leave_type_request_unit == 'day' and check_leave_type:
days = ceil(days)
result[leave.id] = (days, hours)
return result
@api.depends('date_from', 'date_to', 'resource_calendar_id', 'holiday_status_id.request_unit')
def _compute_duration(self):
durations = self._get_durations()
for leave in self:
days, hours = durations[leave.id]
leave.number_of_hours = hours
leave.number_of_days = days
@api.depends('employee_company_id')
def _compute_company_id(self):
for holiday in self:
holiday.company_id = holiday.employee_company_id or holiday.department_id.company_id or self.env.company
@api.depends('number_of_days')
def _compute_last_several_days(self):
for holiday in self:
holiday.last_several_days = holiday.number_of_days > 1
@api.depends('tz')
@api.depends_context('uid')
def _compute_tz_mismatch(self):
for leave in self:
leave.tz_mismatch = leave.tz != self.env.user.tz
@api.depends('resource_calendar_id.tz')
def _compute_tz(self):
for leave in self:
leave.tz = leave.resource_calendar_id.tz or self.env.company.resource_calendar_id.tz or self.env.user.tz or 'UTC'
@api.depends('number_of_hours', 'number_of_days', 'leave_type_request_unit')
def _compute_duration_display(self):
for leave in self:
duration = leave.number_of_days
unit = _('days')
display = "%g %s" % (float_round(duration, precision_digits=2), unit)
if leave.leave_type_request_unit == "hour":
hours, minutes = divmod(abs(leave.number_of_hours) * 60, 60)
minutes = round(minutes)
if minutes == 60:
minutes = 0
hours += 1
duration = '%d:%02d' % (hours, minutes)
unit = _("hours")
display = f"{duration} {unit}"
leave.duration_display = display
@api.depends('state', 'employee_id', 'department_id')
def _compute_can_approve(self):
for holiday in self:
holiday.can_approve = holiday._check_approval_update('validate1', raise_if_not_possible=False)
@api.depends('state', 'employee_id', 'department_id')
def _compute_can_back_to_approve(self):
for holiday in self:
holiday.can_back_to_approve = holiday.state == 'validate' and holiday._check_approval_update('confirm', raise_if_not_possible=False)
@api.depends('state', 'employee_id', 'department_id')
def _compute_can_validate(self):
for holiday in self:
holiday.can_validate = holiday._check_approval_update('validate', raise_if_not_possible=False)
@api.depends('state', 'employee_id', 'department_id')
def _compute_can_refuse(self):
for holiday in self:
holiday.can_refuse = holiday._check_approval_update('refuse', raise_if_not_possible=False)
@api.depends_context('uid')
@api.depends('state', 'employee_id')
def _compute_can_cancel(self):
for holiday in self:
holiday.can_cancel = holiday._check_approval_update('cancel', raise_if_not_possible=False)
@api.depends('state')
def _compute_is_hatched(self):
for holiday in self:
holiday.is_striked = holiday.state == 'refuse'
holiday.is_hatched = holiday.state not in ['refuse', 'validate']
@api.depends('leave_type_support_document', 'attachment_ids')
def _compute_supported_attachment_ids(self):
for holiday in self:
holiday.supported_attachment_ids = holiday.attachment_ids
holiday.supported_attachment_ids_count = len(holiday.attachment_ids.ids)
@api.depends('employee_id', 'holiday_status_id')
def _compute_leaves(self):
date_from = fields.Date.from_string(self.env.context['default_request_date_from']) if 'default_request_date_from' in self.env.context else fields.Date.today()
employee_days_per_allocation = self.employee_id._get_consumed_leaves(self.holiday_status_id, date_from)[0]
for leave in self:
virtual_remaining_leaves = 0
max_leaves = 0
for allocation, allocation_dict in employee_days_per_allocation[leave.employee_id][leave.holiday_status_id].items():
if allocation and (not allocation.date_to or allocation.date_to >= date_from):
max_leaves += allocation_dict['max_leaves']
virtual_remaining_leaves += allocation_dict['virtual_remaining_leaves']
leave.virtual_remaining_leaves = virtual_remaining_leaves
leave.max_leaves = max_leaves
def _inverse_supported_attachment_ids(self):
for holiday in self:
holiday.attachment_ids = holiday.supported_attachment_ids
self.invalidate_recordset(['attachment_ids'])
@api.constrains('date_from', 'date_to', 'employee_id')
def _check_date(self):
if self.env.context.get('leave_skip_date_check', False):
return
for holiday in self:
if holiday.dashboard_warning_message:
raise ValidationError(holiday.dashboard_warning_message)
@api.constrains('date_from', 'date_to', 'employee_id')
def _check_date_state(self):
if self.env.context.get('leave_skip_state_check'):
return
for holiday in self:
if holiday.state in ['validate1', 'validate']:
raise ValidationError(_("This modification is not allowed in the current state."))
def _check_validity(self):
sorted_leaves = defaultdict(lambda: self.env['hr.leave'])
for leave in self:
sorted_leaves[(leave.holiday_status_id, leave.date_from.date())] |= leave
for (leave_type, date_from), leaves in sorted_leaves.items():
if not leave_type.requires_allocation:
continue
employees = leaves.employee_id
leave_data = leave_type.get_allocation_data(employees, date_from)
if leave_type.allows_negative:
max_excess = leave_type.max_allowed_negative
is_cancellation = all(leave.state in ('cancel', 'refuse') for leave in leaves)
for employee in employees:
if is_cancellation:
continue
if not leave_data[employee][0][1]['max_leaves']:
raise ValidationError(_("You do not have any allocation for this time off type.\n"
"Please request an allocation before submitting your time off request."))
if leave_data[employee] and leave_data[employee][0][1]['virtual_remaining_leaves'] < -max_excess:
raise ValidationError(_("There is no valid allocation to cover that request."))
continue
previous_leave_data = leave_type.with_context(
ignored_leave_ids=leaves.ids
).get_allocation_data(employees, date_from)
for employee in employees:
previous_emp_data = previous_leave_data[employee] and previous_leave_data[employee][0][1]['virtual_excess_data']
emp_data = leave_data[employee] and leave_data[employee][0][1]['virtual_excess_data']
if not leave_data[employee][0][1]['max_leaves']:
raise ValidationError(_("You do not have any allocation for this time off type.\n"
"Please request an allocation before submitting your time off request."))
if not previous_emp_data and not emp_data:
continue
if previous_emp_data != emp_data and len(emp_data) >= len(previous_emp_data):
raise ValidationError(_("There is no valid allocation to cover that request."))
is_leave_user = self.env.user.has_group('hr_holidays.group_hr_holidays_user')
if not is_leave_user and any(leave.has_mandatory_day for leave in self):
raise ValidationError(_('You are not allowed to request time off on a Mandatory Day'))
####################################################
# ORM Overrides methods
####################################################
@api.depends(
'tz', 'date_from', 'date_to', 'employee_id',
'holiday_status_id', 'number_of_hours',
'leave_type_request_unit', 'number_of_days', 'department_id',
)
@api.depends_context('short_name', 'hide_employee_name', 'groupby')
def _compute_display_name(self):
for leave in self:
user_tz = pytz.timezone(leave.tz)
date_from_utc = leave.date_from and leave.date_from.astimezone(user_tz).date()
date_to_utc = leave.date_to and leave.date_to.astimezone(user_tz).date()
time_off_type_display = leave.holiday_status_id.name
if self.env.context.get('short_name'):
short_leave_name = leave.name or time_off_type_display or _('Time Off')
leave.display_name = _("%(name)s: %(duration)s", name=short_leave_name, duration=leave.duration_display)
else:
target = leave.employee_id.name or ""
display_date = format_date(self.env, date_from_utc) or ""
if leave.number_of_days > 1 and date_from_utc and date_to_utc:
display_date += _(' to %(date_to_utc)s',
date_to_utc=format_date(self.env, date_to_utc) or ""
)
if not target or self.env.context.get('hide_employee_name') and 'employee_id' in self.env.context.get('group_by', []):
leave.display_name = _("%(leave_type)s: %(duration)s (%(start)s)",
leave_type=time_off_type_display,
duration=leave.duration_display,
start=display_date,
)
elif not time_off_type_display:
leave.display_name = _("%(person)s: %(duration)s (%(start)s)",
person=target,
duration=leave.duration_display,
start=display_date,
)
else:
leave.display_name = _("%(person)s on %(leave_type)s: %(duration)s (%(start)s)",
person=target,
leave_type=time_off_type_display,
duration=leave.duration_display,
start=display_date,
)
def onchange(self, values, field_names, fields_spec):
# Try to force the leave_type display_name when creating new records
# This is called right after pressing create and returns the display_name for
# most fields in the view.
if values and 'employee_id' in fields_spec and 'employee_id' not in self.env.context:
employee_id = get_employee_from_context(values, self.env.context, self.env.user.employee_id.id)
self = self.with_context(employee_id=employee_id)
return super().onchange(values, field_names, fields_spec)
def add_follower(self, employee_id):
employee = self.env['hr.employee'].browse(employee_id)
if employee.user_id:
self.message_subscribe(partner_ids=employee.user_id.partner_id.ids)
def _check_double_validation_rules(self, employees, state):
if self.env.user.has_group('hr_holidays.group_hr_holidays_manager'):
return
is_leave_user = self.env.user.has_group('hr_holidays.group_hr_holidays_user')
if state == 'validate1':
employees = employees.filtered(lambda employee: employee.leave_manager_id != self.env.user)
if employees and not is_leave_user:
raise AccessError(_('You cannot first approve a time off for %s, because you are not his time off manager', employees[0].name))
elif state == 'validate' and not is_leave_user:
# Is probably handled via ir.rule
raise AccessError(_('You don\'t have the rights to apply second approval on a time off request'))
@api.model_create_multi
def create(self, vals_list):
# Override to avoid automatic logging of creation
if not self.env.context.get('leave_fast_create'):
leave_types = self.env['hr.leave.type'].browse([values.get('holiday_status_id') for values in vals_list if values.get('holiday_status_id')])
mapped_validation_type = {leave_type.id: leave_type.leave_validation_type for leave_type in leave_types}
for values in vals_list:
employee_id = values.get('employee_id', False)
leave_type_id = values.get('holiday_status_id')
# Handle double validation
if mapped_validation_type[leave_type_id] == 'both':
self._check_double_validation_rules(employee_id, values.get('state', False))
if any(not vals.get('employee_id') for vals in vals_list):
raise UserError(_("There is no employee set on the time off. Please make sure you're logged in the correct company."))
holidays = super(HrLeave, self.with_context(mail_create_nosubscribe=True)).create(vals_list)
holidays._check_validity()
self.env['hr.leave.allocation'].invalidate_model(['leaves_taken', 'max_leaves']) # missing dependency on compute
for holiday in holidays:
if not self.env.context.get('leave_fast_create'):
# Everything that is done here must be done using sudo because we might
# have different create and write rights
# eg : holidays_user can create a leave request with validation_type = 'manager' for someone else
# but they can only write on it if they are leave_manager_id
holiday_sudo = holiday.sudo()
holiday_sudo.add_follower(holiday.employee_id.id)
if holiday.validation_type == 'manager':
holiday_sudo.message_subscribe(partner_ids=holiday.employee_id.leave_manager_id.partner_id.ids)
if holiday.validation_type == 'no_validation':
# Automatic validation should be done in sudo, because user might not have the rights to do it by himself
holiday_sudo.action_approve()
holiday_sudo.message_subscribe(partner_ids=holiday._get_responsible_for_approval().partner_id.ids)
holiday_sudo.message_post(body=_("The time off has been automatically approved"), subtype_xmlid="mail.mt_comment") # Message from OdooBot (sudo)
elif not self.env.context.get('import_file'):
holiday_sudo.activity_update()
return holidays
def write(self, vals):
values = vals
is_officer = self.env.user.has_group('hr_holidays.group_hr_holidays_user') or self.env.is_superuser()
if not is_officer and values.keys() - {'attachment_ids', 'supported_attachment_ids', 'message_main_attachment_id'}:
if any(hol.date_from.date() < fields.Date.today() and hol.employee_id.leave_manager_id != self.env.user
and hol.state not in ('confirm', 'draft') for hol in self):
raise UserError(_('You must have manager rights to modify/validate a time off that already begun'))
if any(leave.state == 'cancel' for leave in self):
raise UserError(_('Only a manager can modify a canceled leave.'))
# Unlink existing resource.calendar.leaves for validated time off
if 'state' in values and values['state'] != 'validate':
validated_leaves = self.filtered(lambda l: l.state == 'validate')
validated_leaves._remove_resource_leave()
employee_id = values.get('employee_id', False)
if not self.env.context.get('leave_fast_create'):
if values.get('state'):
self._check_approval_update(values['state'])
if any(holiday.validation_type == 'both' for holiday in self):
if values.get('employee_id'):
employees = self.env['hr.employee'].browse(values.get('employee_id'))
else:
employees = self.mapped('employee_id')
self._check_double_validation_rules(employees, values['state'])
if 'date_from' in values:
values['request_date_from'] = values['date_from']
if 'date_to' in values:
values['request_date_to'] = values['date_to']
result = super().write(values)
if any(field in values for field in ['request_date_from', 'date_from', 'request_date_from', 'date_to', 'holiday_status_id', 'employee_id', 'state']):
if not values.get('state') or values.get('state') not in ('refuse', 'cancel'):
self._check_validity()
self.env['hr.leave.allocation'].invalidate_model(['leaves_taken', 'max_leaves']) # missing dependency on compute
if not self.env.context.get('leave_fast_create'):
for holiday in self:
if employee_id:
holiday.add_follower(employee_id)
return result
@api.ondelete(at_uninstall=False)
def _unlink_if_correct_states(self):
error_message = self.env._('Oops! %(state)s Time-Off requests can only be deleted by Administrators.')
state_description_values = {elem[0]: elem[1] for elem in self._fields['state']._description_selection(self.env)}
now = fields.Datetime.now().date()
if not self.env.user.has_group('hr_holidays.group_hr_holidays_user'):
for hol in self:
if hol.state not in ['confirm', 'validate1', 'cancel']:
raise UserError(error_message % {'state': state_description_values.get(self[:1].state)})
if hol.date_from.date() < now:
raise UserError(_("You can't delete a time off request that is in the past."))
elif not self.env.user.has_group('hr_holidays.group_hr_holidays_manager'):
for holiday in self.filtered(lambda holiday: holiday.state not in ['cancel', 'confirm']):
error_message = self.env._('Oops! %(state)s Time-Off requests can only be deleted by Administrators.')
raise UserError(error_message % {'state': state_description_values.get(holiday.state)})
def unlink(self):
self.sudo()._post_leave_cancel()
self.env['hr.leave.allocation'].invalidate_model(['leaves_taken', 'max_leaves']) # missing dependency on compute
return super(HrLeave, self.with_context(leave_skip_date_check=True)).unlink()
def copy_data(self, default=None):
vals_list = super().copy_data(default=default)
if self.env.context.get('skip_copy_check'):
return vals_list
if all(leave.state in ['cancel', 'refuse'] for leave in self): # No overlap constraint in these cases
return vals_list
raise UserError(_('A time off cannot be duplicated.'))
def _get_redirect_suggested_company(self):
return self.holiday_status_id.company_id
####################################################
# Business methods
####################################################
def _prepare_resource_leave_vals(self):
"""Hook method for others to inject data
"""
self.ensure_one()
return {
'name': _("%s: Time Off", self.employee_id.name),
'date_from': self.date_from,
'holiday_id': self.id,
'date_to': self.date_to,
'resource_id': self.employee_id.resource_id.id,
'calendar_id': self.resource_calendar_id.id,
'time_type': self.holiday_status_id.time_type,
'elligible_for_accrual_rate': self.holiday_status_id.elligible_for_accrual_rate,
}
def _create_resource_leave(self):
""" This method will create entry in resource calendar time off object at the time of holidays validated
:returns: created `resource.calendar.leaves`
"""
vals_list = [leave._prepare_resource_leave_vals() for leave in self]
return self.env['resource.calendar.leaves'].sudo().create(vals_list)
def _remove_resource_leave(self):
""" This method will create entry in resource calendar time off object at the time of holidays cancel/removed """
if self.has_access('write'):
return self.env['resource.calendar.leaves'].search([('holiday_id', 'in', self.ids)]).sudo().unlink()
return self.env['resource.calendar.leaves'].search([('holiday_id', 'in', self.ids)]).unlink()
def _validate_leave_request(self):
""" Validate time off requests
by creating a calendar event and a resource time off. """
holidays = self.filtered("employee_id")
holidays._create_resource_leave()
meeting_holidays = holidays.filtered(lambda l: l.holiday_status_id.create_calendar_meeting)
meetings = self.env['calendar.event']
if meeting_holidays:
Meeting = self.env['calendar.event']
Meeting.check_access('create')
meeting_values_for_user_id = meeting_holidays._prepare_holidays_meeting_values()
Meeting = self.env['calendar.event']
for user_id, meeting_values in meeting_values_for_user_id.items():
meetings += Meeting.with_user(user_id or self.env.uid).sudo().with_context(clean_context({**self.env.context, **dict(
allowed_company_ids=[],
no_mail_to_attendees=True,
calendar_no_videocall=True,
active_model=self._name
)})).create(meeting_values)
Holiday = self.env['hr.leave']
for meeting in meetings:
Holiday.browse(meeting.res_id).meeting_id = meeting
for holiday in holidays:
user_tz = pytz.timezone(holiday.tz)
utc_tz = pytz.utc.localize(holiday.date_from).astimezone(user_tz)
notify_partner_ids = holiday.employee_id.user_id.partner_id.ids
holiday.message_post(
body=_(
'Your %(leave_type)s planned on %(date)s has been accepted',
leave_type=holiday.holiday_status_id.display_name,
date=utc_tz.replace(tzinfo=None)
),
partner_ids=notify_partner_ids)
def _prepare_holidays_meeting_values(self):
result = defaultdict(list)
for holiday in self:
user = holiday.user_id
meeting_name = _(
"%(employee)s on Time Off : %(duration)s",
employee=holiday.employee_id.name or holiday.category_id.name,
duration=holiday.duration_display)
allday_value = not holiday.request_unit_half or holiday.request_date_from_period == 'am' and holiday.request_date_to_period == 'pm'
if holiday.leave_type_request_unit == 'hour':
allday_value = float_compare(holiday.number_of_days, 1.0, 1) >= 0
if allday_value:
# `start` and `stop` are not in UTC for allday events
leave_tz = pytz.timezone(holiday.tz) if holiday.tz else pytz.UTC
start_value = pytz.UTC.localize(holiday.date_from).astimezone(leave_tz).replace(tzinfo=None)
stop_value = pytz.UTC.localize(holiday.date_to).astimezone(leave_tz).replace(tzinfo=None)
else:
start_value = holiday.date_from
stop_value = holiday.date_to
meeting_values = {
'name': meeting_name,
'duration': holiday.number_of_days * (holiday.resource_calendar_id.hours_per_day or HOURS_PER_DAY),
'description': holiday.notes,
'user_id': user.id,
'start': start_value,
'stop': stop_value,
'allday': allday_value,
'privacy': 'confidential',
'event_tz': user.tz,
'activity_ids': [(5, 0, 0)],
'res_id': holiday.id,
}
# Add the partner_id (if exist) as an attendee
partner_id = (user and user.partner_id) or (holiday.employee_id and holiday.employee_id.work_contact_id)
if partner_id:
meeting_values['partner_ids'] = [(4, partner_id.id)]
result[user.id].append(meeting_values)
return result
def action_cancel(self):
self.ensure_one()
return {
'name': _('Cancel Time Off'),
'type': 'ir.actions.act_window',
'target': 'new',
'res_model': 'hr.holidays.cancel.leave',
'view_mode': 'form',
'views': [[False, 'form']],
'context': {
'default_leave_id': self.id,
'dialog_size': "medium",
}
}
def action_approve(self, check_state=True):
current_employee = self.env.user.employee_id
leave_to_approve = self.env['hr.leave']
leave_to_validate = self.env['hr.leave']
for leave in self:
if check_state and leave.can_validate or not check_state and leave.validation_type != "both":
leave_to_validate += leave
elif check_state and leave.can_approve or not check_state and leave.validation_type == 'both':
leave_to_approve += leave
else:
raise UserError(self.env._('You cannot approve this leave.'))
leave_to_approve.write({'state': 'validate1', 'first_approver_id': current_employee.id})
leave_to_validate._action_validate(check_state)
if not self.env.context.get('leave_fast_create'):
self.activity_update()
return True
def action_back_to_approval(self):
self.filtered(lambda l: l.can_back_to_approve)._move_validate_leave_to_confirm()
return True
def _move_validate_leave_to_confirm(self):
self.write({'state': 'confirm'})
self.activity_update()
self._post_leave_cancel()
def _get_leaves_on_public_holiday(self):
return self.filtered(lambda l: l.employee_id and not l.number_of_days)
def _split_leaves(self, split_date_from, split_date_to=False):
"""
This method splits an original leave in two leaves and returns the new one for each leave in self.
E.g. (start, stop) -> (start, split_date_from - 1day), (split_date_to, stop)
:param split_date_from: The starting date of the splicing interval (includes)
:param split_date_to: The ending date of the splicing interval. (not includes)
:param changes_message: The message will be translated and posted in the first leave's chatter
If split_date_to is not set; the splicing interval will be equals to [split_date_form, split_date_from -1]
to avoid one day leave.
"""
new_leaves_vals = []
if not split_date_to:
split_date_to = split_date_from
# Only leaves that span a period outside of the split interval need
# to be split.
multi_day_leaves = self.filtered(lambda l: l.request_date_from < split_date_from or l.request_date_to >= split_date_to)
for leave in multi_day_leaves:
new_leave_vals = []
target_leave_vals = []
if leave.request_date_from < split_date_from:
new_leave_vals.append(leave.with_context(skip_copy_check=True).copy_data({
'request_date_to': split_date_from + timedelta(days=-1),
'state': leave.state
})[0])
# Do the same for the new leave after the split
if leave.request_date_to >= split_date_to:
new_leave_vals.append(leave.with_context(skip_copy_check=True).copy_data({
'request_date_from': split_date_to,
'state': leave.state
})[0])
# For those two new leaves, only create them if they actually have a non-zero duration.
for leave_vals in new_leave_vals:
new_leave = self.env['hr.leave'].new(leave_vals)
new_leave._compute_date_from_to()
if new_leave.date_from < new_leave.date_to:
target_leave_vals.append(new_leave._convert_to_write(new_leave._cache))
if target_leave_vals:
vals = target_leave_vals.pop(0)
leave.with_context(leave_skip_state_check=True).write({
'request_date_from': vals['request_date_from'],
'request_date_to': vals['request_date_to'],
})
if target_leave_vals:
new_leaves_vals.extend(target_leave_vals)
if not new_leaves_vals:
return self.env['hr.leave']
return self.env['hr.leave'].with_context(
tracking_disable=True,
mail_activity_automation_skip=True,
leave_fast_create=True,
leave_skip_state_check=True
).create(new_leaves_vals)
def _action_validate(self, check_state=True):
current_employee = self.env.user.employee_id
leaves = self._get_leaves_on_public_holiday()
if check_state and any(not holiday.can_validate for holiday in self):
raise UserError(_('You can\'t validate this leave.'))
if leaves:
raise ValidationError(_('The following employees are not supposed to work during that period:\n %s') % ','.join(leaves.mapped('employee_id.name')))
self.write({'state': 'validate'})
leaves_second_approver = self.env['hr.leave']
leaves_first_approver = self.env['hr.leave']
for leave in self:
if leave.validation_type == 'both':
leaves_second_approver += leave
else:
leaves_first_approver += leave
leaves_second_approver.write({'second_approver_id': current_employee.id})
leaves_first_approver.write({'first_approver_id': current_employee.id})
self._validate_leave_request()
if not self.env.context.get('leave_fast_create'):
self.filtered(lambda holiday: holiday.validation_type != 'no_validation').activity_update()
return True
def action_refuse(self):
current_employee = self.env.user.employee_id
if any(holiday.state not in ['confirm', 'validate', 'validate1'] for holiday in self):
raise UserError(_('Time off request must be confirmed or validated in order to refuse it.'))
self._notify_manager()
validated_holidays = self.filtered(lambda hol: hol.state == 'validate1')
validated_holidays.write({'state': 'refuse', 'first_approver_id': current_employee.id})
(self - validated_holidays).write({'state': 'refuse', 'second_approver_id': current_employee.id})
# Delete the meeting
self.mapped('meeting_id').write({'active': False})
# Post a second message, more verbose than the tracking message
for holiday in self:
if holiday.employee_id.user_id:
holiday.message_post(
body=_('Your %(leave_type)s planned on %(date)s has been refused', leave_type=holiday.holiday_status_id.display_name, date=holiday.date_from),
partner_ids=holiday.employee_id.user_id.partner_id.ids)
self.activity_update()
return True
def _notify_manager(self):
leaves = self.filtered(lambda hol: (hol.validation_type == 'both' and hol.state in ['validate1', 'validate']) or (hol.validation_type == 'manager' and hol.state == 'validate'))
model_description = self.env['ir.model']._get('hr.holidays').name
for holiday in leaves:
responsible = holiday.employee_id.leave_manager_id.partner_id.ids
if responsible:
holiday.sudo().message_notify(
partner_ids=responsible,
model_description=model_description,
subject=_('Refused Time Off'),
body=_(
'%(holiday_name)s has been refused.',
holiday_name=holiday.display_name,
),
email_layout_xmlid="mail.mail_notification_layout",
subtitles=[holiday.display_name],
)
def _action_user_cancel(self, reason=None):
self.ensure_one()
if not self.can_cancel:
raise ValidationError(_('This time off cannot be cancelled.'))
self._force_cancel(reason, 'mail.mt_note')
def _force_cancel(self, reason=None, msg_subtype='mail.mt_comment', notify_responsibles=True):
leaves = self.browse() if self.env.context.get(MODULE_UNINSTALL_FLAG) else self
if reason:
model_description = self.env['ir.model']._get('hr.holidays').display_name
for leave in leaves:
body = self.env._(
"The time off request has been cancelled for the following reason:%(reason)s",
reason=Markup("{reason}
").format(reason=reason)
)
leave.message_post(
body=body,
subtype_xmlid=msg_subtype
)
if not notify_responsibles:
continue
responsibles = self.env['res.partner']
# manager
if (leave.holiday_status_id.leave_validation_type == 'manager' and leave.state == 'validate') or (leave.holiday_status_id.leave_validation_type == 'both' and leave.state == 'validate1'):
responsibles = leave.employee_id.leave_manager_id.partner_id
# officer
elif leave.holiday_status_id.leave_validation_type == 'hr' and leave.state == 'validate':
responsibles = leave.holiday_status_id.responsible_ids.partner_id
# both
elif leave.holiday_status_id.leave_validation_type == 'both' and leave.state == 'validate':
responsibles = leave.employee_id.leave_manager_id.partner_id
responsibles |= leave.holiday_status_id.responsible_ids.partner_id
if responsibles:
body = self.env._(
"%(leave_name)s has been cancelled for the following reason: %(reason)s",
leave_name=leave.display_name,
reason=Markup("{reason}
").format(reason=reason),
)
leave.message_notify(
partner_ids=responsibles.ids,
model_description=model_description,
subject=self.env._('Cancelled Time Off'),
body=body,
email_layout_xmlid="mail.mail_notification_layout",
subtitles=[leave.display_name],
)
leave_sudo = self.sudo()
leave_sudo.state = "cancel"
leave_sudo.activity_update()
leave_sudo._post_leave_cancel()
def _post_leave_cancel(self):
self.meeting_id.active = False
self._remove_resource_leave()
def action_documents(self):
domain = [('id', 'in', self.attachment_ids.ids)]
return {
'name': _("Supporting Documents"),
'type': 'ir.actions.act_window',
'res_model': 'ir.attachment',
'context': {'create': False},
'view_mode': 'kanban',
'domain': domain
}
def _get_next_states_by_state(self):
self.ensure_one()
state_result = {
'confirm': set(),
'validate1': set(),
'validate': set(),
'refuse': set(),
'cancel': set()
}
validation_type = self.validation_type
user_employees = self.env.user.employee_ids
is_own_leave = self.employee_id in user_employees
is_in_past = self.date_from and self.date_from.date() < fields.Date.today()
is_officer = self.env.user.has_group('hr_holidays.group_hr_holidays_user')
is_time_off_manager = self.employee_id.leave_manager_id == self.env.user
if is_own_leave and (not is_in_past or is_officer):
state_result['validate1'].add('cancel')
state_result['validate'].add('cancel')
state_result['refuse'].add('cancel')
if is_officer:
if validation_type == 'both':
state_result['confirm'].add('validate1')
state_result['refuse'].add('validate1')
state_result['cancel'].add('validate1')
state_result['confirm'].update({'validate', 'refuse'})
state_result['validate1'].update({'confirm', 'validate', 'refuse'})
state_result['validate'].update({'confirm', 'refuse'})
state_result['refuse'].update({'confirm', 'validate'})
state_result['cancel'].update({'confirm', 'validate', 'refuse'})
elif is_time_off_manager:
if validation_type != 'hr':
state_result['confirm'].add('refuse')
state_result['validate'].add('refuse')
if validation_type == 'both':
state_result['confirm'].add('validate1')
state_result['validate1'].add('refuse')
elif validation_type == 'manager':
state_result['confirm'].add('validate')
state_result['refuse'].add('validate')
return state_result
def _check_approval_update(self, state, raise_if_not_possible=True):
""" Check if target state is achievable. """
if self.env.is_superuser():
return True
is_officer = self.env.user.has_group('hr_holidays.group_hr_holidays_user')
for holiday in self:
is_time_off_manager = holiday.employee_id.leave_manager_id == self.env.user
dict_all_possible_state = holiday._get_next_states_by_state()
validation_type = holiday.validation_type
error_message = ""
# Standard Check
if holiday.state == state:
error_message = self.env._('You can\'t do the same action twice.')
elif state == 'validate1' and validation_type != 'both':
error_message = self.env._('Not possible state. State Approve is only used for leave needed 2 approvals')
elif holiday.state == 'cancel':
error_message = self.env._('A cancelled leave cannot be modified.')
elif state not in dict_all_possible_state.get(holiday.state, {}):
if state == 'cancel':
error_message = self.env._('You can only cancel your own leave. You can cancel a leave only if this leave \
is approved, validated or refused.')
elif state == 'confirm':
error_message = self.env._('You can\'t reset a leave. Cancel/delete this one and create an other')
elif state == 'validate1':
if not is_time_off_manager:
error_message = self.env._('Only a Time Off Officer/Manager can approve a leave.')
else:
error_message = self.env._('You can\'t approve a validated leave.')
elif state == "validate":
if not is_time_off_manager:
error_message = self.env._('Only a Time Off Officer/Manager can validate a leave.')
elif holiday.state == "refuse":
error_message = self.env._('You can\'t approve this refused leave.')
else:
error_message = self.env._('You can only validate a leave with validation by Time Off Manager.')
elif state == "refuse":
if not is_time_off_manager:
error_message = self.env._('Only a Time Off Officer/Manager can refuse a leave.')
else:
error_message = self.env._('You can\'t refuse a leave with validation by Time Off Officer.')
elif state != "cancel":
try:
holiday.check_access('write')
except UserError as e:
if raise_if_not_possible:
raise UserError(e)
return False
else:
continue
if error_message:
if raise_if_not_possible:
raise UserError(error_message)
return False
return True
@api.model
def open_pending_requests(self):
user_employee = self.env.user.employee_id
employee = self.env['hr.employee']._get_contextual_employee()
context = {'search_default_approve': True, 'search_default_second_approval': True}
domain = []
if employee != user_employee:
view_name = 'hr_holidays.hr_leave_allocation_view_tree'
context.update({'search_default_employee_id': employee.id})
else:
view_name = 'hr_holidays.hr_leave_allocation_view_tree_my'
domain = [('employee_id', '=', employee.id)]
return {
'name': _('Allocation Requests'),
'type': 'ir.actions.act_window',
'res_model': 'hr.leave.allocation',
'views': [[self.env.ref(view_name).id, 'list']],
'domain': domain,
'context': context,
}
# ------------------------------------------------------------
# Activity methods
# ------------------------------------------------------------
def _get_responsible_for_approval(self):
self.ensure_one()
responsible = self.env['res.users']
if self.validation_type == 'manager' or (self.validation_type == 'both' and self.state == 'confirm'):
if self.employee_id.leave_manager_id:
responsible = self.employee_id.leave_manager_id
elif self.employee_id.parent_id.user_id:
responsible = self.employee_id.parent_id.user_id
elif self.validation_type == 'hr' or (self.validation_type == 'both' and self.state == 'validate1'):
if self.holiday_status_id.responsible_ids:
responsible = self.holiday_status_id.responsible_ids
return responsible
def _get_to_clean_activities(self):
return ['hr_holidays.mail_act_leave_approval', 'hr_holidays.mail_act_leave_second_approval']
def activity_update(self):
if self.env.context.get('mail_activity_automation_skip'):
return
to_clean, to_do, to_do_confirm_activity = self.env['hr.leave'], self.env['hr.leave'], self.env['hr.leave']
activity_vals = []
today = fields.Date.today()
model_id = self.env['ir.model']._get_id('hr.leave')
confirm_activity = self.env.ref('hr_holidays.mail_act_leave_approval')
approval_activity = self.env.ref('hr_holidays.mail_act_leave_second_approval')
for holiday in self:
if holiday.state in ['confirm', 'validate1']:
if holiday.holiday_status_id.leave_validation_type != 'no_validation':
if holiday.state == 'confirm':
activity_type = confirm_activity
note = _(
'New %(leave_type)s Request created by %(user)s',
leave_type=holiday.holiday_status_id.name,
user=holiday.create_uid.name,
)
else:
activity_type = approval_activity
note = _(
'Second approval request for %(leave_type)s',
leave_type=holiday.holiday_status_id.name,
)
to_do_confirm_activity += holiday
user_ids = holiday.sudo()._get_responsible_for_approval().ids
for user_id in user_ids:
date_deadline = (
(holiday.date_from -
relativedelta(**{activity_type.delay_unit or 'days': activity_type.delay_count or 0})).date()
if holiday.date_from else today)
if date_deadline < today:
date_deadline = today
activity_vals.append({
'activity_type_id': activity_type.id,
'automated': True,
'date_deadline': date_deadline,
'note': note,
'user_id': user_id,
'res_id': holiday.id,
'res_model_id': model_id,
})
elif holiday.state == 'validate':
to_do |= holiday
elif holiday.state in ['refuse', 'cancel']:
to_clean |= holiday
if to_clean:
to_clean.activity_unlink(self._get_to_clean_activities(), only_automated=False)
if to_do_confirm_activity:
to_do_confirm_activity.activity_feedback(['hr_holidays.mail_act_leave_approval'])
if to_do:
to_do.activity_feedback(['hr_holidays.mail_act_leave_approval', 'hr_holidays.mail_act_leave_second_approval'])
self.env['mail.activity'].with_context(short_name=False).create(activity_vals)
####################################################
# Messaging methods
####################################################
def _notify_change(self, message, subtype_xmlid='mail.mt_note'):
for leave in self:
leave.message_post(body=message, subtype_xmlid=subtype_xmlid)
recipient = None
if leave.user_id:
recipient = leave.user_id.partner_id.id
elif leave.employee_id:
recipient = leave.employee_id.work_contact_id.id
if recipient:
self.env['mail.thread'].sudo().message_notify(
body=message,
partner_ids=[recipient],
subject=_('Your Time Off'),
)
def _track_subtype(self, init_values):
if 'state' in init_values and self.state == 'validate':
leave_notif_subtype = self.holiday_status_id.leave_notif_subtype_id
return leave_notif_subtype or self.env.ref('hr_holidays.mt_leave')
return super()._track_subtype(init_values)
def message_subscribe(self, partner_ids=None, subtype_ids=None):
# due to record rule can not allow to add follower and mention on validated leave so subscribe through sudo
if any(holiday.state in ['validate', 'validate1'] for holiday in self):
self.check_access('read')
return super(HrLeave, self.sudo()).message_subscribe(partner_ids=partner_ids, subtype_ids=subtype_ids)
return super().message_subscribe(partner_ids=partner_ids, subtype_ids=subtype_ids)
@api.model
def get_unusual_days(self, date_from, date_to=None):
employee_id = self.env.context.get('employee_id', False)
employee = self.env['hr.employee'].browse(employee_id) if employee_id else self.env.user.employee_id
return employee.sudo(False)._get_unusual_days(date_from, date_to)
def _to_utc(self, date, hour, resource):
hour = float_to_time(float(hour))
holiday_tz = pytz.timezone(resource.tz or self.env.user.tz or 'UTC')
return holiday_tz.localize(datetime.combine(date, hour)).astimezone(pytz.UTC).replace(tzinfo=None)
def _get_hour_from_to(self, request_date_from, request_date_to, day_period=None):
"""
Return the hour_from and hour_to for the given request dates, based on
the resource calendar.
If there are no attendances on the exact days of the request, return
the earliest hour_from and latest hour_to that exist in the schedule.
"""
calendar = self.resource_calendar_id
if not calendar:
return (0, 24)
calendar.ensure_one()
hour_from, _ = calendar._get_hours_for_date(request_date_from, day_period)
_, hour_to = calendar._get_hours_for_date(request_date_to, day_period)
return (hour_from, hour_to)
####################################################
# Cron methods
####################################################
@api.model
def _cancel_invalid_leaves(self):
inspected_date = fields.Date.today() + timedelta(days=31)
start_datetime = datetime.combine(fields.Date.today(), datetime.min.time())
end_datetime = datetime.combine(inspected_date, datetime.max.time())
concerned_leaves = self.search([
('date_from', '>=', start_datetime),
('date_from', '<=', end_datetime),
('state', 'in', ['confirm', 'validate1', 'validate']),
], order='date_from desc')
accrual_allocations = self.env['hr.leave.allocation'].search([
('employee_id', 'in', concerned_leaves.employee_id.ids),
('holiday_status_id', 'in', concerned_leaves.holiday_status_id.ids),
('allocation_type', '=', 'accrual'),
('date_from', '<=', end_datetime),
'|',
('date_to', '>=', start_datetime),
('date_to', '=', False),
])
# only take leaves linked to accruals
concerned_leaves = concerned_leaves\
.filtered(lambda leave: leave.holiday_status_id in accrual_allocations.holiday_status_id)\
.sorted('date_from', reverse=True)
reason = _("the accruated amount is insufficient for that duration.")
for leave in concerned_leaves:
leave_type = leave.holiday_status_id
date = leave.date_from.date()
leave_type_data = leave_type.get_allocation_data(leave.employee_id, date)
if not leave_type_data[leave.employee_id][0][1]['max_leaves']:
leave._force_cancel(reason, 'mail.mt_note')
continue
exceeding_duration = leave_type_data[leave.employee_id][0][1]['total_virtual_excess']
excess_limit = leave_type.max_allowed_negative if leave_type.allows_negative else 0
if exceeding_duration <= excess_limit:
continue
leave._force_cancel(reason, 'mail.mt_note')