Add constraints for maximum days and unique shifts per week block in worker model

This commit is contained in:
Ross
2026-05-19 21:23:09 +01:00
parent a071b1b027
commit dee80ed4fa
5 changed files with 192 additions and 4 deletions
+3 -1
View File
@@ -15,7 +15,7 @@ from rota_generator.workers import (
NonWorkingDays,
WorkRequests,
PreferenceNotToWork,
OutOfProgramme,
OutOfProgramme, MaxUniqueShiftsPerWeekBlockConstraint,
)
from loguru import logger
@@ -465,6 +465,8 @@ def load_workers():
else:
w.add_hard_day_dependency({"Fri", "Sun"})
w.add_hard_day_exclusion({"Sat", "Sun"})
else:
w.max_unique_shifts_per_week_block = [MaxUniqueShiftsPerWeekBlockConstraint(week_block=1, max_unique_shifts=1)]