from typing import Any

from app.algorithms.valve_isolation.topology_search import valve_isolation_analysis
from app.native.wndb.gis.network_views import get_network_link_nodes


def analyze_valve_isolation(
    network: str,
    accident_element: str | list[str],
    disabled_valves: list[str] = None,
) -> dict[str, Any]:
    link_entries = get_network_link_nodes(network)
    return valve_isolation_analysis(link_entries, accident_element, disabled_valves)