From ab426bb6e74ca6cfff29d840726940a16df1b3ba Mon Sep 17 00:00:00 2001 From: Cizz22 Date: Thu, 27 Nov 2025 11:59:55 +0700 Subject: [PATCH] add status to workflow --- temporal/workflow.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/temporal/workflow.py b/temporal/workflow.py index cfbd664..3ae54d9 100644 --- a/temporal/workflow.py +++ b/temporal/workflow.py @@ -7,9 +7,17 @@ from temporal.activity import calculate_plant_eaf_activity, execute_simulation_a @workflow.defn class SimulationWorkflow: + status = "PENDING" + + @workflow.query + def get_status(self) -> str: + return self.status + @workflow.run async def run(self, sim_data: dict) -> str: - # 1. Update equipment + self.status = "RUNNING" + + # 1. Update equipment for simulation results = await workflow.execute_activity( update_equipment_for_simulation_activity, {**sim_data}, # ✅ one positional argument @@ -36,5 +44,6 @@ class SimulationWorkflow: start_to_close_timeout=timedelta(days=1) ) + self.status = "COMPLETED" # etc… return sim_data["HubCnnId"] # simulation_id