def _update_future_states()

in azure-slurm/slurmcc/cli.py [0:0]

• 22 lines of code
• 7 McCabe index (conditional complexity)

def _update_future_states(node_mgr: NodeManager, node_list: Optional[List[str]] = None) -> None:
    autoscale_enabled = is_autoscale_enabled()
    if autoscale_enabled:
        return
    nodes = node_mgr.get_nodes()

    for node in nodes:
        if node_list and node.name not in node_list:
            continue

        if node.target_state != "Started":
            name = node.name
            try:
                cmd = [
                    "scontrol",
                    "update",
                    f"NodeName={name}",
                    f"NodeAddr={name}",
                    f"NodeHostName={name}",
                    "state=FUTURE",
                ]
                check_output(cmd)
            except SubprocessError:
                logging.warning(f"Could not set {node.get('Name')} state=FUTURE")