def get_task_status()

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


def get_task_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:
        log.info(f"Task {taskid} completed")
        return taskset_handler.TaskStatus.COMPLETED
    elif task.state == batchmodels.TaskState.running:
        log.info(f"Task {taskid} is running")
        return taskset_handler.TaskStatus.RUNNING

    log.info(f"Task {taskid} state={task.state} UNKNOWN")
    return taskset_handler.TaskStatus.UNKNOWN