make sure pre and post scripts are added to send/receive tasks

sartography · Jul 2, 2024 · 7e15a0b · 7e15a0b
1 parent 7392ec6
commit 7e15a0b
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/SpiffWorkflow/spiff/parser/event_parsers.py b/SpiffWorkflow/spiff/parser/event_parsers.py
@@ -162,8 +162,18 @@ def create_task(self):
 
 class SpiffSendTaskParser(SpiffEventDefinitionParser, SendTaskParser):
     def create_task(self):
-        return SendTaskParser.create_task(self)
+        task = SendTaskParser.create_task(self)
+        # Duplicate this code from the SpiffTask parser because of parser limitations
+        extensions = self.parse_extensions()
+        task.prescript = extensions.get('preScript')
+        task.postscript = extensions.get('postScript')
+        return task
 
 class SpiffReceiveTaskParser(SpiffEventDefinitionParser, ReceiveTaskParser):
     def create_task(self):
-        return ReceiveTaskParser.create_task(self)
+        task = ReceiveTaskParser.create_task(self)
+        # I have given up trying to do any of this right
+        extensions = self.parse_extensions()
+        task.prescript = extensions.get('preScript')
+        task.postscript = extensions.get('postScript')
+        return task