import networkx as nx
from .utils import _build_graph_ip_only
def path(device, target_device_name: str) -> dict:
"""
Compute shortest path between two devices.
"""
topology_result = _build_graph_ip_only()
topology_graph = topology_result.graph
source_device_name = device.name if hasattr(device, "name") else device
try:
path_nodes = nx.shortest_path(
topology_graph,
source=source_device_name,
target=target_device_name,
)
path_edges = []
for idx in range(len(path_nodes) - 1):
from_device, to_device = path_nodes[idx], path_nodes[idx + 1]
path_edges.append(
{
"source": from_device,
"target": to_device,
"attributes": topology_graph.get_edge_data(
from_device, to_device
),
}
)
return {"nodes": path_nodes, "edges": path_edges}
except Exception as error:
return {"error": str(error)}
