Update rota configuration for February 2026 and refine shift constraints handling

This commit is contained in:
Ross
2025-12-18 22:06:31 +00:00
parent 3ba4467812
commit 1bae1246f9
2 changed files with 14 additions and 12 deletions
+4 -2
View File
@@ -482,7 +482,7 @@ def main(
weeks_to_rota=weeks,
balance_offset_modifier=bom,
use_previous_shifts=True,
name="cons rota feb 2026 run 1",
name="cons rota feb 2026 run 2",
allow_force_assignment_with_leave_conflict=True,
constraint_options=RotaConstraintOptions(
balance_weekends=True,
@@ -510,7 +510,8 @@ def main(
assign_as_block=False,
constraints=[
PreShiftConstraint(days=2, start_date="2025-11-17", exclude_days=("Fri", "Sat", "Sun"), exclude_dates=["2026-04-06", "2026-05-04", "2026-05-25"]),
PostShiftConstraint(days=1, start_date="2025-11-17", exclude_days=("Fri", "Sat", "Sun"), allow_self=True),
PreShiftConstraint(days=1, start_date="2025-11-17", exclude_days=("Fri", "Sat", "Sun"), allow_self=False),
MaxShiftsPerWeekConstraint(max_shifts=1, days=days[:5]),
],
),
SingleShift(
@@ -529,6 +530,7 @@ def main(
SingleShift(
sites=("rota b", "rota c"),
name="weekend b",
end_date="2026-07-20",
workers_required=1,
length=12.5,
days=days[5:],
+10 -10
View File
@@ -126,6 +126,7 @@ class RequireRemoteSitePresenceConstraint(BaseShiftConstraint):
class MaxShiftsPerWeekConstraint(BaseShiftConstraint):
max_shifts: int | None = None
days: Optional[List[str]] = None
class MaxShiftsPerWeekBlockConstraint(BaseShiftConstraint):
week_block : int
@@ -2788,20 +2789,19 @@ class RotaBuilder(object):
shift_number = constraint.max_shifts
else:
shift_number = constraint_shift.get_shift_number()
try:
# If the constraint specifies particular days, only count those days,
# otherwise count the shift's own days.
constraint_days = (
constraint.days if constraint.days is not None else self.days
)
self.model.constraints.add(
shift_number
>= sum(
self.model.works[
worker.id, w, day, constraint_shift.name
]
# for shiftname in self.get_shift_names_by_week_day(week, day)
# for day in self.days
for w, day in self.get_week_day_combinations_for_shift(
constraint_shift
)
if w == week
self.model.works[worker.id, w, day, constraint_shift.name]
for w, day in self.get_week_day_combinations_for_shift(constraint_shift)
if w == week and day in constraint_days
)
)
except ValueError: