def wait_task_completion()

in src/hpcadvisor/batch_handler.py [0:0]

• 14 lines of code
• 6 McCabe index (conditional complexity)

def wait_task_completion(jobid, taskid, wait_blocked=True):
    log.info(f"Waiting for task completion {taskid}")

    if not batch_client:
        log.critical("batch_client is None")
        return

    # TODO: need to add a maximum amount of time for task completion
    while True:
        task = batch_client.task.get(jobid, taskid)
        log.info(f"task state={task.state} task id={taskid}")
        if task.state == batchmodels.TaskState.completed:
            break
        if not wait_blocked:
            return False
        time.sleep(5)

    return True