update replacement to subassembly id to avoid killing every process o…

…n a system

wombat/core/repair_management.py

@@ -533,7 +533,7 @@ def invalidate_system(
             )
 
     def interrupt_system(
-        self, system: System | Cable, replacement: bool = False
+        self, system: System | Cable, replacement: str | None = None
     ) -> None:
         """Sets the turbine status to be in servicing, and interrupts all the processes
         to turn off operations.
@@ -542,9 +542,9 @@ def interrupt_system(
         ----------
         system_id : str
             The system to disable repairs.
-        replacement: bool, optional
-            Indicates if the interruption is caused a result of a replacement event.
-            Defaults to False.
+        replacement: str | None, optional
+            If a subassebly `id` is provided, this indicates the interruption is caused
+            by its replacement event. Defaults to None.
         """
         if system.servicing.triggered and system.id in self.invalid_systems:
             system.servicing = self.env.event()

wombat/core/service_equipment.py

@@ -1471,9 +1471,10 @@ def in_situ_repair(
             raise RuntimeError(f"{self.settings.name} is lost!")
 
         if initial:
-            self.manager.interrupt_system(
-                system, replacement=request.details.replacement
+            replacement = (
+                request.subassembly_id if request.details.replacement else None
             )
+            self.manager.interrupt_system(system, replacement=replacement)
         yield self.env.process(
             self.crew_transfer(system, subassembly, request, to_system=True)
         )
@@ -1891,7 +1892,8 @@ def run_tow_to_port(self, request: RepairRequest) -> Generator[Process, None, No
         )
 
         # Turn off the turbine
-        self.manager.interrupt_system(system, replacement=request.details.replacement)
+        replacement = request.subassembly_id if request.details.replacement else None
+        self.manager.interrupt_system(system, replacement=replacement)
 
         # Unmoor the turbine and tow it back to port
         yield self.env.process(self.mooring_connection(system, request, which="unmoor"))

wombat/windfarm/system/cable.py

@@ -169,7 +169,7 @@ def recreate_processes(self) -> None:
         """
         self.processes = dict(self._create_processes())
 
-    def interrupt_processes(self, replacement: bool = False) -> None:
+    def interrupt_processes(self, replacement: str | None = None) -> None:
         """Interrupts all of the running processes within the subassembly except for the
         process associated with failure that triggers the catastrophic failure.
 
@@ -178,25 +178,30 @@ def interrupt_processes(self, replacement: bool = False) -> None:
         subassembly : Subassembly
             The subassembly that should have all processes interrupted.
         replacement: bool, optional
-            Indicates if the interruption is caused a result of a replacement event.
-            Defaults to False.
+            If a subassebly `id` is provided, this indicates the interruption is caused
+            by its replacement event. Defaults to None.
         """
-        cause = "replacement" if replacement else "failure"
+        cause = "failure"
+        if self.id == replacement:
+            cause = "replacement"
+
         for _, process in self.processes.items():
             try:
                 process.interrupt(cause=cause)
             except RuntimeError:
                 # This error occurs for the process halting all other processes.
                 pass
 
-    def interrupt_all_subassembly_processes(self, replacement: bool = False) -> None:
+    def interrupt_all_subassembly_processes(
+        self, replacement: str | None = None
+    ) -> None:
         """Thin wrapper for ``interrupt_processes`` for consistent usage with system.
 
         Parameters
         ----------
         replacement: bool, optional
-            Indicates if the interruption is caused a result of a replacement event.
-            Defaults to False.
+            If a subassebly `id` is provided, this indicates the interruption is caused
+            by its replacement event. Defaults to None.
         """
         self.interrupt_processes(replacement=replacement)
 

wombat/windfarm/system/subassembly.py

@@ -84,7 +84,7 @@ def recreate_processes(self) -> None:
         self.processes = dict(self._create_processes())
 
     def interrupt_processes(
-        self, origin: Subassembly | None = None, replacement: bool = False
+        self, origin: Subassembly | None = None, replacement: str | None = None
     ) -> None:
         """Interrupts all of the running processes within the subassembly except for the
         process associated with failure that triggers the catastrophic failure.
@@ -97,10 +97,12 @@ def interrupt_processes(
             current subassembly, then a try/except flow is used to ensure the process
             that initiated the shutdown is not interrupting itself.
         replacement: bool, optional
-            Indicates if the interruption is caused a result of a replacement event.
-            Defaults to False.
+            If a subassebly `id` is provided, this indicates the interruption is caused
+            by its replacement event. Defaults to None.
         """
-        cause = "replacement" if replacement else "failure"
+        cause = "failure"
+        if self.id == replacement:
+            cause = "replacement"
         if origin is not None and id(origin) == id(self):
             for _, process in self.processes.items():
                 try:

wombat/windfarm/system/system.py

@@ -162,7 +162,7 @@ def _initialize_power_curve(self, power_curve_dict: dict | None) -> None:
             )
 
     def interrupt_all_subassembly_processes(
-        self, origin: Subassembly | None = None, replacement: bool = False
+        self, origin: Subassembly | None = None, replacement: str | None = None
     ) -> None:
         """Interrupts the running processes in all of the system's subassemblies.
 
@@ -172,8 +172,8 @@ def interrupt_all_subassembly_processes(
             The subassembly that triggered the request, if the method call is coming
             from a subassembly shutdown event.
         replacement: bool, optional
-            Indicates if the interruption is caused a result of a replacement event.
-            Defaults to False.
+            If a subassebly `id` is provided, this indicates the interruption is caused
+            by its replacement event. Defaults to None.
         """
         [
             subassembly.interrupt_processes(origin=origin, replacement=replacement)  # type: ignore