chore: add register_signal option to ScenarioExecution for thread

IntelLabs · Oct 16, 2024 · 1f788c7 · 1f788c7
1 parent 80d5489
commit 1f788c7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/scenario_execution/scenario_execution/scenario_execution_base.py b/scenario_execution/scenario_execution/scenario_execution_base.py
@@ -92,13 +92,15 @@ def __init__(self,
                  setup_timeout=py_trees.common.Duration.INFINITE,
                  tick_period: float = 0.1,
                  scenario_parameter_file=None,
-                 logger=None) -> None:
+                 logger=None,
+                 register_signal=True) -> None:
 
         def signal_handler(sig, frame):
             self.on_scenario_shutdown(False, "Aborted")
 
-        signal.signal(signal.SIGHUP, signal_handler)
-        signal.signal(signal.SIGTERM, signal_handler)
+        if register_signal:
+            signal.signal(signal.SIGHUP, signal_handler)
+            signal.signal(signal.SIGTERM, signal_handler)
 
         self.current_scenario_start = None
         self.current_scenario = None