update command types to add Payload class

Author: bitterpanda63

aikido_zen/background_process/commands/command.py

@@ -17,7 +17,24 @@ class Command(ABC):
     def identifier(cls) -> str:
         pass
 
+    @classmethod
+    @abstractmethod
+    def returns_data(cls) -> bool:
+        pass
+
     @classmethod
     @abstractmethod
     def run(cls, context: CommandContext, request):
         pass
+
+    @classmethod
+    @abstractmethod
+    def generate(cls, *args, **kwargs):
+        pass
+
+
+class Payload:
+    def __init__(self, command: type[Command], request):
+        self.identifier = command.identifier()
+        self.returns_data = command.returns_data()
+        self.request = request

aikido_zen/background_process/commands/put_event.py

@@ -1,4 +1,4 @@
-from .command import Command, CommandContext
+from .command import Command, CommandContext, Payload
 
 
 class PutEventReq:
@@ -13,6 +13,14 @@ class PutEventCommand(Command):
     def identifier(cls) -> str:
         return "put_event"
 
+    @classmethod
+    def returns_data(cls) -> bool:
+        return False
+
     @classmethod
     def run(cls, context: CommandContext, request: PutEventReq):
         context.queue.put(request.event)
+
+    @classmethod
+    def generate(cls, event) -> Payload:
+        return Payload(cls, PutEventReq(event))