Put a single CalcJob with a transport request

aiidateam · Oct 30, 2024 · 7785525 · 7785525
1 parent 7057238
commit 7785525
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/src/aiida/engine/runners.py b/src/aiida/engine/runners.py
@@ -252,16 +252,21 @@ def kill_process(_num, _frame):
                 LOGGER.critical('runner received interrupt, killing process %s', process_inited.pid)
                 process_inited.kill(msg='Process was killed because the runner received an interrupt')
 
-            original_handler_int = signal.getsignal(signal.SIGINT)
-            original_handler_term = signal.getsignal(signal.SIGTERM)
+            # TODO: only do this if the node is CalcJob
+            authinfo = process_inited.node.get_authinfo()
 
-            try:
-                signal.signal(signal.SIGINT, kill_process)
-                signal.signal(signal.SIGTERM, kill_process)
-                process_inited.execute()
-            finally:
-                signal.signal(signal.SIGINT, original_handler_int)
-                signal.signal(signal.SIGTERM, original_handler_term)
+            with self.transport.request_transport(authinfo):
+
+                original_handler_int = signal.getsignal(signal.SIGINT)
+                original_handler_term = signal.getsignal(signal.SIGTERM)
+
+                try:
+                    signal.signal(signal.SIGINT, kill_process)
+                    signal.signal(signal.SIGTERM, kill_process)
+                    process_inited.execute()
+                finally:
+                    signal.signal(signal.SIGINT, original_handler_int)
+                    signal.signal(signal.SIGTERM, original_handler_term)
 
             return process_inited.outputs, process_inited.node