From de5ae4d1e5d30dc70103e8b7f4d1173f01fb57fb Mon Sep 17 00:00:00 2001
From: DingZQ <ding@aecforge.com>
Date: Sun, 8 Dec 2024 21:08:06 +0800
Subject: [PATCH] Add API get network link nodes

---
 main.py      |  7 +++++++
 tjnetwork.py | 10 ++++++++++
 2 files changed, 17 insertions(+)

diff --git a/main.py b/main.py
index e3ca326..b87e1cb 100644
--- a/main.py
+++ b/main.py
@@ -1391,6 +1391,7 @@ async def fastapi_set_option_properties(network: str, req: Request) -> ChangeSet
 async def fastapi_get_node_coord(network: str, node: str) -> dict[str, float] | None:
     return get_node_coord(network, node)
 
+# DingZQ, 2024-12-08, get all node coord
 @app.get("/getnetworkcoords/")
 async def fastapi_get_network_coords(network: str) -> list[str] | None:
     coords = get_network_coords(network)
@@ -1399,6 +1400,12 @@ async def fastapi_get_network_coords(network: str) -> list[str] | None:
         result.append(f"{node_id}:{coord['x']}:{coord['y']}")
     return result
 
+# DingZQ, 2024-12-08, get all links' start and end node
+# link_id:node_id1:node_id2
+@app.get("/getnetworklinknodes/")
+async def fastapi_get_network_link_nodes(network: str) -> list[str] | None:
+    return get_network_link_nodes(network)
+
 ############################################################
 # vertex 25.[VERTICES]
 ############################################################
diff --git a/tjnetwork.py b/tjnetwork.py
index c1c56de..c820a01 100644
--- a/tjnetwork.py
+++ b/tjnetwork.py
@@ -839,6 +839,16 @@ def get_network_coords(name: str) -> dict[str, dict[str, float]]:
         result[node_id] = api.get_node_coord(name, node_id)
     return result
 
+# DingZQ, 2024-12-08, get all links' start and end node
+# link_id:node_id1:node_id2
+def get_network_link_nodes(name: str) -> list[str]:
+    links = api.get_links(name)
+    result = []
+    for link_id in links:
+        nodes = api.get_link_nodes(name, link_id)
+        result.append(f"{link_id}:{nodes[0]}:{nodes[1]}")
+    return result
+
 ############################################################
 # vertex 25.[VERTICES]
 ############################################################