def get_task_execution_status()

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


def get_task_execution_status(jobname, taskid):
    if batch_client is None:
        log.critical("batch_client is None")
        return taskset_handler.TaskStatus.UNKNOWN

    task = batch_client.task.get(jobname, taskid)
    if (
        task.state == batchmodels.TaskState.completed
        and task.execution_info.exit_code != 0
    ):
        log.info(f"Task {taskid} completed but failed")
        return taskset_handler.TaskStatus.FAILED
    elif (
        task.state == batchmodels.TaskState.completed
        and task.execution_info.exit_code == 0
    ):
        log.info(f"Task {taskid} completed without error")
        return taskset_handler.TaskStatus.COMPLETED

    log.warning(f"Task {taskid} unkown state")
    return taskset_handler.TaskStatus.UNKNOWN