add flowcept

Author: danielrosendo

wfcommons/wfbench/translator/pycompss.py

@@ -46,8 +46,13 @@ def translate(self, output_folder: pathlib.Path) -> None:
         :type output_folder: pathlib.Path
         """
         self.output_folder = output_folder
-        self.script = "\n# workflow tasks\n"
+        self.script = ""
+        # IMPORT Flowcept
+        if self.workflow.workflow_id is not None:
+            self.script += "from flowcept.flowcept_api.flowcept_controller import Flowcept\n\n"
+
         # PyCOMPSs translator
+        self.script += "\n# workflow tasks\n"
         self._pycompss_code()
 
         # Generates pycompss workflow file: template + script
@@ -199,4 +204,12 @@ def _pycompss_code(self) -> None:
 
         # CALL TO MAIN METHOD
         self.script += f"\n\nif __name__ == \"__main__\":\n"
+        # START Flowcept
+        if self.workflow.workflow_id is not None:
+            self.script += f"\tf = Flowcept(workflow_id='{self.workflow.workflow_id}', workflow_name='{self.workflow.name}', bundle_exec_id='{self.workflow.workflow_id}')\n"
+            self.script += "\tf.start()\n"
+        # main
         self.script += f"\tmain_program()\n"
+        # STOP Flowcept
+        if self.workflow.workflow_id is not None:
+            self.script += "\tf.stop()\n"

wfcommons/wfbench/translator/templates/pycompss_template.py

@@ -1,32 +1,11 @@
-# IMPORTS
-import sys
 import os
-# PyCOMPSs imports
-# All task arguments: https://compss-doc.readthedocs.io/en/stable/Sections/02_App_Development/02_Python/01_1_Task_definition/Sections/04_Task_parameters_summary.html
 from pycompss.api.task import task
 from pycompss.api.constraint import constraint
 from pycompss.api.binary import binary
 from pycompss.api.mpi import mpi
 from pycompss.api.parameter import *
-# All API functions: https://compss-doc.readthedocs.io/en/stable/Sections/02_App_Development/02_Python/01_2_Synchronization/01_API.html?highlight=compss_open#api-summary
 from pycompss.api.api import compss_open
 from pycompss.api.api import compss_wait_on
 from pycompss.api.api import compss_barrier
 
-# @binary
-# @constraint(computing_units=24)
-# @mpi(runner="mpirun", binary="gmx_mpi", computing_nodes=1)
-# @task
-# def task_id():
-#     pass
-#
-# def main_program(arg1, arg2):
-#     # Execute task
-#     task_id()
-#
-# if __name__ == "__main__":
-#     arg1 = sys.argv[1]
-#     arg2 = sys.argv[2]
-#     main_program(arg1, arg2)
-
 # Generated code goes here