From 5256243c762bc1176b37c7c2d38a696b9506ecb5 Mon Sep 17 00:00:00 2001 From: DingZQ Date: Mon, 29 Aug 2022 21:29:34 +0800 Subject: [PATCH] Add code to get link nodes --- main.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/main.py b/main.py index 3edd2d5..fbcf344 100644 --- a/main.py +++ b/main.py @@ -135,6 +135,12 @@ async def fastapi_get_links(network: str): return links +@app.get("/getlinknodes/") +async def fastapi_get_link_nodes(network: str, link: str): + result = get_link_nodes(network, link) + + return [result.value.from_node, result.value.to_node] + @app.post("/addlink/") async def fastapi_add_link(network: str, link: str, fromNode: str, toNode: str): idx = add_link(network, link, PIPE, fromNode, toNode)