bf5c391e47
Signed-off-by: JoshuaL3000 <joshua.jian.ern.liew@intel.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import time
|
|
|
|
from tools.sdk import EasyDataSDK
|
|
|
|
|
|
def workflow_executor(params, workflow_id: int) -> dict:
|
|
sdk = EasyDataSDK()
|
|
workflow = sdk.create_workflow(workflow_id)
|
|
|
|
params = {key: str(val) for key, val in params.items()}
|
|
start_workflow = workflow.start(params)
|
|
print(start_workflow)
|
|
|
|
def check_workflow():
|
|
workflow_status = workflow.get_status()["workflow_status"]
|
|
if workflow_status == "finished":
|
|
message = "Workflow finished."
|
|
elif workflow_status == "initializing" or workflow_status == "running":
|
|
message = "Workflow execution is still in progress."
|
|
else:
|
|
message = "Workflow has failed."
|
|
|
|
return workflow_status, message
|
|
|
|
MAX_RETRY = 50
|
|
num_retry = 0
|
|
while num_retry < MAX_RETRY:
|
|
workflow_status, message = check_workflow()
|
|
print(message)
|
|
if workflow_status == "failed" or workflow_status == "finished":
|
|
break
|
|
else:
|
|
time.sleep(100) # interval between each status checking retry
|
|
num_retry += 1
|
|
|
|
if workflow_status == "finished":
|
|
return workflow.result()
|
|
else:
|
|
return message
|