From 43a3aba7bfc541671a1c8b1aeb8579248f4abb5d Mon Sep 17 00:00:00 2001 From: Cizz22 Date: Thu, 9 Oct 2025 13:08:36 +0700 Subject: [PATCH] fix --- src/calculation_target_reliability/router.py | 22 ++++++++++++++++++++ src/calculation_time_constrains/router.py | 3 +++ 2 files changed, 25 insertions(+) diff --git a/src/calculation_target_reliability/router.py b/src/calculation_target_reliability/router.py index 9aa39f9..fc15abb 100644 --- a/src/calculation_target_reliability/router.py +++ b/src/calculation_target_reliability/router.py @@ -63,6 +63,28 @@ async def get_target_reliability( sim_hours=duration, token=token ) + + workflow_id = f"simulation-{simulation_id}" + + while True: + try: + handle = temporal_client.get_workflow_handle(workflow_id=workflow_id) + desc = await handle.describe() + status = desc.status.name + + if status not in ["RUNNING", "CONTINUED_AS_NEW"]: + print(f"Workflow {workflow_id} finished with status: {status}") + return status + + print(f"Workflow {workflow_id} still {status}, checking again in {interval} seconds...") + + except WorkflowNotFoundError: + print(f"Workflow {workflow_id} not found, treating as done.") + return "NOT_FOUND" + + await asyncio.sleep(interval) + + results = await get_simulation_results( simulation_id=simulation_id, diff --git a/src/calculation_time_constrains/router.py b/src/calculation_time_constrains/router.py index 7c61772..252a047 100644 --- a/src/calculation_time_constrains/router.py +++ b/src/calculation_time_constrains/router.py @@ -139,6 +139,9 @@ async def update_selected_equipment( calculation_id, calculation_time_constrains_in: List[CalculationSelectedEquipmentUpdate], ): + if calculation_id == 'default': + calculation_id = "3b9a73a2-bde6-418c-9e2f-19046f501a05" + results = await bulk_update_equipment( db=db_session, selected_equipments=calculation_time_constrains_in,