Function parse_exec_line

tests/autoalloc/test_autoalloc.py:428–454

`script_path` should be a path to qsub or sbatch submit script.

(script_path: str)

def parse_exec_line(script_path: str) -> WorkerExecLine:
    """
    `script_path` should be a path to qsub or sbatch submit script.
    """
    line = get_exec_line(script_path)

    def get_env(item: str) -> Optional[Tuple[str, str]]:
        if "=" in item:
            key, value = item.split("=", maxsplit=1)
            if key.isupper():
                return (key, value)
        return None

    env = {}
    cmd = None
    items = line.split(" ")
    for index, item in enumerate(items):
        env_item = get_env(item)
        if env_item is not None:
            assert env_item[0] not in env
            env[env_item[0]] = env_item[1]
        else:
            cmd = " ".join(items[index:])
            break

    assert cmd is not None
    return WorkerExecLine(cmd=cmd, env=env)


@all_flavors