Catch sigterm as well

not just catch ctrl+c but also sigterms coming from CI builds stopping the docker container running this

Author: FlorentinD

scripts/ci/run_targeting_aura_sessions.py

@@ -24,13 +24,14 @@ def main() -> None:
     instance_id = create_result["id"]
     logging.info(f"Creation of database with id '{instance_id}'")
 
-    # Teardown instance on SIGINT
-    def handle_sigint(sig, frame):
-        logging.info("Received SIGINT, tearing down instance")
+    # Teardown instance on SIGNAL
+    def handle_signal(sig, frame):
+        logging.info("Received SIGNAL, tearing down instance")
         aura_api.teardown_instance(instance_id)
         sys.exit(1)
 
-    signal.signal(signal.SIGINT, handle_sigint)
+    signal.signal(signal.SIGINT, handle_signal)
+    signal.signal(signal.SIGTERM, handle_signal)
 
     try:
         aura_api.check_running(instance_id)

scripts/ci/run_targeting_aurads.py

@@ -39,12 +39,13 @@ def main() -> None:
     instance_id = create_result["id"]
     logging.info("Creation of database accepted")
 
-    def handle_sigint(sig, frame):
-        logging.info("Received SIGINT, tearing down instance")
+    def handle_signal(sig, frame):
+        logging.info("Received SIGNAL, tearing down instance")
         aura_api.teardown_instance(instance_id)
         sys.exit(1)
 
-    signal.signal(signal.SIGINT, handle_sigint)
+    signal.signal(signal.SIGINT, handle_signal)
+    signal.signal(signal.SIGTERM, handle_signal)
 
     try:
         aura_api.check_running(instance_id)

scripts/run_notebooks.py

@@ -32,12 +32,13 @@ def init_notebook(self, version_cell_index: int, tear_down_cells: List[IndexedCe
     def preprocess_cell(self, cell: Any, resources: Any, index: int) -> None:
         if index == 0:
 
-            def handle_sigint(sig, frame):
-                print("Received SIGINT, running tear down cells")
+            def handle_signal(sig, frame):
+                print("Received SIGNAL, running tear down cells")
                 self.teardown(resources)
                 sys.exit(1)
 
-            signal.signal(signal.SIGINT, handle_sigint)
+            signal.signal(signal.SIGINT, handle_signal)
+            signal.signal(signal.SIGTERM, handle_signal)
 
         try:
             if not self._skip_rest: