1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from datetime import datetime
from pytz import timezone, utc
from odoo import models
class ResourceCalendarLeaves(models.Model):
_inherit = 'resource.calendar.leaves'
def _compute_calendar_id(self):
def date2datetime(date, tz):
dt = datetime.fromordinal(date.toordinal())
return tz.localize(dt).astimezone(utc).replace(tzinfo=None)
leaves_by_contract = self.grouped(lambda leave: leave.resource_id.employee_id.contract_id)
# set aside leaves without contract_id for super
remaining = leaves_by_contract.pop(
self.env['hr.contract'],
self.env['resource.calendar.leaves'],
)
for contract, leaves in leaves_by_contract.items():
tz = timezone(contract.resource_calendar_id.tz or 'UTC')
start_dt = date2datetime(contract.date_start, tz)
end_dt = date2datetime(contract.date_end, tz) if contract.date_end else datetime.max
# only modify leaves that fall under the active contract
leaves.filtered(
lambda leave: start_dt <= leave.date_from < end_dt
).calendar_id = contract.resource_calendar_id
super(ResourceCalendarLeaves, remaining)._compute_calendar_id()
|
