add status to workflow

main
Cizz22 2 months ago
parent 5a626c3d67
commit ab426bb6e7

@ -7,9 +7,17 @@ from temporal.activity import calculate_plant_eaf_activity, execute_simulation_a
@workflow.defn @workflow.defn
class SimulationWorkflow: class SimulationWorkflow:
status = "PENDING"
@workflow.query
def get_status(self) -> str:
return self.status
@workflow.run @workflow.run
async def run(self, sim_data: dict) -> str: async def run(self, sim_data: dict) -> str:
# 1. Update equipment self.status = "RUNNING"
# 1. Update equipment for simulation
results = await workflow.execute_activity( results = await workflow.execute_activity(
update_equipment_for_simulation_activity, update_equipment_for_simulation_activity,
{**sim_data}, # ✅ one positional argument {**sim_data}, # ✅ one positional argument
@ -36,5 +44,6 @@ class SimulationWorkflow:
start_to_close_timeout=timedelta(days=1) start_to_close_timeout=timedelta(days=1)
) )
self.status = "COMPLETED"
# etc… # etc…
return sim_data["HubCnnId"] # simulation_id return sim_data["HubCnnId"] # simulation_id

Loading…
Cancel
Save