Add default MESSAGE_KEY to None value and FORCE_KILL_KEY

unkcpz · Dec 1, 2024 · 7732647 · 7732647
1 parent 21508e3
commit 7732647
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/plumpy/process_comms.py b/src/plumpy/process_comms.py
@@ -31,7 +31,7 @@
 
 INTENT_KEY = 'intent'
 MESSAGE_KEY = 'message'
-
+FORCE_KILL_KEY = 'force_kill'
 
 class Intent:
     """Intent constants for a process message"""
@@ -42,10 +42,10 @@ class Intent:
     STATUS: str = 'status'
 
 
-PAUSE_MSG = {INTENT_KEY: Intent.PAUSE}
-PLAY_MSG = {INTENT_KEY: Intent.PLAY}
-KILL_MSG = {INTENT_KEY: Intent.KILL}
-STATUS_MSG = {INTENT_KEY: Intent.STATUS}
+PAUSE_MSG = {INTENT_KEY: Intent.PAUSE, MESSAGE_KEY: None}
+PLAY_MSG = {INTENT_KEY: Intent.PLAY, MESSAGE_KEY: None}
+KILL_MSG = {INTENT_KEY: Intent.KILL, MESSAGE_KEY: None, FORCE_KILL_KEY: False}
+STATUS_MSG = {INTENT_KEY: Intent.STATUS, MESSAGE_KEY: None}
 
 TASK_KEY = 'task'
 TASK_ARGS = 'args'