18.0-ref #20

Open
laetitiadacosta wants to merge 3 commits from 18.0-ref into 18.0
3 changed files with 216 additions and 56 deletions
Showing only changes of commit bf27d0938c - Show all commits

View File

@@ -3,7 +3,7 @@
{
"name": "hr_holidays_timeoff_analysis",
"version": "18.0.1.1.0",
"version": "18.0.2.0.0",
"author": "Elabore",
"website": "https://git.elabore.coop/elabore/elabore-addons",
"maintainer": "Elabore",

View File

@@ -137,12 +137,18 @@ class TimeOffDay(models.Model):
@api.model
def cron_manage_timeoff_days(self):
self.cron_create_timeoff_days()
self.cron_delete_timeoff_days()
"""Reconcile hr.leave.timeoff.day records day by day.
def cron_create_timeoff_days(self):
# Browse all validated leaves
leaves = self.env["hr.leave"].search(
On each day within the relevant date range, compute which timeoff.days
*should* exist (validated leaves covering the day, employee scheduled
to work, not a public holiday) and reconcile against existing records:
- Stale records (leave dates changed, state changed, force-cancelled,
orphan, calendar/holiday change) are deleted.
- Missing records are created.
- Existing records with an outdated duration are updated in place.
"""
# Load all currently validated leaves that have dates and an employee.
valid_leaves = self.env["hr.leave"].search(
[
("state", "=", "validate"),
("request_date_from", "!=", False),
@@ -150,59 +156,104 @@ class TimeOffDay(models.Model):
("employee_id", "!=", False),
]
)
for leave in leaves:
current_date = leave.request_date_from
employee = leave.employee_id
while current_date <= leave.request_date_to:
# If no validated leaves exist, all existing timeoff.days are stale/orphan.
if not valid_leaves:
self.search([]).unlink()
return
# Determine the full date span to iterate over: from the earliest
# relevant date (oldest leave or oldest existing timeoff.day) to the
# latest relevant date (newest leave or newest existing timeoff.day).
#first, get the oldest and latest leave dates
leave_from_dates = valid_leaves.mapped("request_date_from")
leave_to_dates = valid_leaves.mapped("request_date_to")
start_date = min(leave_from_dates)
end_date = max(leave_to_dates)
#then, get oldest timeoff.day and latest timeoff.day
#and keep as start_date and end_date the min and max of both
oldest_td = self.search([], order="date asc", limit=1)
if oldest_td and oldest_td.date and oldest_td.date < start_date:
start_date = oldest_td.date
latest_td = self.search([], order="date desc", limit=1)
if latest_td and latest_td.date and latest_td.date > end_date:
end_date = latest_td.date
existing_tds = self.search(
[("date", ">=", start_date), ("date", "<=", end_date)]
)
# Build in-memory indexes for fast lookup per day.
# leaves_by_date: date -> list of validated leaves covering that day
leaves_by_date = {}
for leave in valid_leaves:
d = leave.request_date_from
while d <= leave.request_date_to:
if d not in leaves_by_date:
leaves_by_date[d] = []
leaves_by_date[d].append(leave)
d += timedelta(days=1)
# existing_by_date: date -> list of existing timeoff.day records
existing_by_date = {}
for td in existing_tds:
if td.date not in existing_by_date:
existing_by_date[td.date] = []
existing_by_date[td.date].append(td)
# Collect operations and apply them at the end to minimise DB round trips.
all_to_create = []
all_to_delete_ids = []
current_date = start_date
while current_date <= end_date:
covering_leaves = leaves_by_date.get(current_date, [])
# Compute the expected timeoff.days for this date.
# Key: (employee_id, leave_id) -> expected leave_duration_by_day
expected = {}
for leave in covering_leaves:
employee = leave.employee_id
if self.employee_is_scheduled_to_work_this_day(
current_date, employee, leave
) and not self._is_public_holiday_according_to_employe_tz(
current_date, employee
):
# The employee is scheduled to work this day
# according his calendar and it's not a
# public holiday,
# so create a timeoff day record if it does not already exist
if not self.search(
[
("date", "=", current_date),
("employee_id", "=", employee.id),
("hr_leave_id", "=", leave.id),
],
limit=1,
):
self.create(
{
"date": current_date,
"employee_id": employee.id,
"hr_leave_id": leave.id,
"leave_duration_by_day": (
self.compute_leave_duration_by_day(leave)
),
}
)
current_date += timedelta(days=1)
expected[(employee.id, leave.id)] = (
self.compute_leave_duration_by_day(leave)
)
def cron_delete_timeoff_days(self):
# Browse all unvalidated or canceled leaves
leaves = self.env["hr.leave"].search(
[
("state", "!=", "validate"),
("request_date_from", "!=", False),
("request_date_to", "!=", False),
("employee_id", "!=", False),
]
)
# Delete timeoff days for leaves that are no longer validated
for leave in leaves:
self.search(
[
("hr_leave_id", "=", leave.id),
]
).unlink()
# Delete timeoff days that are not linked to any leave
self.search(
[
("hr_leave_id", "=", False),
]
).unlink()
# Reconcile existing timeoff.days for this date.
existing_today = existing_by_date.get(current_date, [])
for td in existing_today:
key = (td.employee_id.id, td.hr_leave_id.id)
if key in expected:
# Matched: keep the record, update duration if it changed.
expected_duration = expected.pop(key)
if td.leave_duration_by_day != expected_duration:
td.leave_duration_by_day = expected_duration
else:
# No longer valid: leave dates changed, state changed,
# force-cancelled, orphan (hr_leave_id=False), calendar
# changed, or the day became a public holiday.
all_to_delete_ids.append(td.id)
# Remaining entries in expected are missing timeoff.days -> create.
for (employee_id, leave_id), duration in expected.items():
all_to_create.append(
{
"date": current_date,
"employee_id": employee_id,
"hr_leave_id": leave_id,
"leave_duration_by_day": duration,
}
)
current_date += timedelta(days=1)
# Apply batched operations.
if all_to_delete_ids:
self.browse(all_to_delete_ids).unlink()
if all_to_create:
self.create(all_to_create)

View File

@@ -717,3 +717,112 @@ class TestHrLeaveTimeoffDay(TransactionCase):
4,
"leave_duration_by_day should be 4 day",
)
def test_leave_dates_modified(self):
leave = self.env["hr.leave"].create(
{
"employee_id": self.employee.id,
"request_date_from": Date.to_date("2025-09-01"), # Monday
"request_date_to": Date.to_date("2025-09-05"), # Friday
"holiday_status_id": self.time_off_type.id,
}
)
leave.state = "validate"
self.env["hr.leave.timeoff.day"].cron_manage_timeoff_days()
timeoff_days = self.env["hr.leave.timeoff.day"].search(
[
("employee_id", "=", self.employee.id),
("hr_leave_id", "=", leave.id),
]
)
self.assertEqual(
len(timeoff_days), 5, "Should have 5 timeoff days initially"
)
# Modify dates: Wed Sept 3 to Sun Sept 7
# Working days: Wed(3), Thu(4), Fri(5) — Mon(1), Tue(2) removed
leave.write(
{
"request_date_from": Date.to_date("2025-09-03"),
"request_date_to": Date.to_date("2025-09-07"),
}
)
self.env["hr.leave.timeoff.day"].cron_manage_timeoff_days()
timeoff_days = self.env["hr.leave.timeoff.day"].search(
[
("employee_id", "=", self.employee.id),
("hr_leave_id", "=", leave.id),
]
)
self.assertEqual(
len(timeoff_days),
3,
"Should have 3 timeoff days after date modification",
)
updated_dates = timeoff_days.mapped("date")
self.assertIn(
Date.to_date("2025-09-03"),
updated_dates,
"Sept 3 should still be present",
)
self.assertIn(
Date.to_date("2025-09-04"),
updated_dates,
"Sept 4 should still be present",
)
self.assertIn(
Date.to_date("2025-09-05"),
updated_dates,
"Sept 5 should still be present",
)
self.assertNotIn(
Date.to_date("2025-09-01"),
updated_dates,
"Sept 1 should have been removed",
)
self.assertNotIn(
Date.to_date("2025-09-02"),
updated_dates,
"Sept 2 should have been removed",
)
def test_leave_duration_changed(self):
leave = self.env["hr.leave"].create(
{
"employee_id": self.employee.id,
"request_date_from": Date.to_date("2025-10-01"),
"request_date_to": Date.to_date("2025-10-01"),
"holiday_status_id": self.time_off_type.id,
}
)
leave.state = "validate"
self.env["hr.leave.timeoff.day"].cron_manage_timeoff_days()
timeoff_day = self.env["hr.leave.timeoff.day"].search(
[
("employee_id", "=", self.employee.id),
("hr_leave_id", "=", leave.id),
]
)
self.assertEqual(len(timeoff_day), 1)
self.assertEqual(
timeoff_day.leave_duration_by_day,
1.0,
"Should be 1.0 for a full day leave",
)
# Simulate a stale duration, then verify the cron corrects it
# in place (exercises the update-in-place reconciliation path).
timeoff_day.leave_duration_by_day = 0.5
self.env["hr.leave.timeoff.day"].cron_manage_timeoff_days()
timeoff_day = self.env["hr.leave.timeoff.day"].search(
[
("employee_id", "=", self.employee.id),
("hr_leave_id", "=", leave.id),
]
)
self.assertEqual(len(timeoff_day), 1)
self.assertEqual(
timeoff_day.leave_duration_by_day,
1.0,
"Should be corrected back to 1.0 for a full day leave",
)