a lot of changes

rota/workers.py

@@ -114,6 +114,10 @@ class Worker(BaseModel):
     allowed_multi_shift_sets: list[set] = []  # or list/set of frozensets
     prefer_multi_shift_together: int = 0  # Default: no preference
     # Set to a positive integer to prefer, negative to discourage, 0 for neutral
+    hard_day_dependencies: dict[str, str] = {}  # e.g. {"Fri": "Sun"}
+    hard_day_exclusions: list[tuple[str, str]] = []  # e.g. [("Fri", "Sun")]
+
+    assign_as_block_preferences: dict[str, float] = {}  # shift_name -> weight (positive = prefer block)
 
     shift_fte_overrides: dict[str, int] = {} # Need checks to ensure shifts exist
 
@@ -396,4 +400,14 @@ class Worker(BaseModel):
     def allow_shifts_together(self, *shifts):
         if not hasattr(self, "allowed_multi_shift_sets"):
             self.allowed_multi_shift_sets = []
-        self.allowed_multi_shift_sets.append(frozenset(shifts))
+        self.allowed_multi_shift_sets.append(frozenset(shifts))
+
+    def add_hard_day_dependency(self, from_day: str, to_day: str):
+        if not hasattr(self, "hard_day_dependencies"):
+            self.hard_day_dependencies = {}
+        self.hard_day_dependencies[from_day] = to_day
+
+    def add_hard_day_exclusion(self, day1: str, day2: str):
+        if not hasattr(self, "hard_day_exclusions"):
+            self.hard_day_exclusions = []
+        self.hard_day_exclusions.append((day1, day2))