OrgTJWater/TJWaterServer
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add diameter to getmajornodes/getmajorpipes

Browse Source
This commit is contained in:
DingZQ
2025-01-01 10:19:12 +08:00
parent 51e3ae18bb
commit 9e7a4616b4
3 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@@ -1407,8 +1407,8 @@ async def fastapi_get_network_coords(network: str) -> list[str] | None:
# id:type:x:y
# type: junction, reservoir, tank
@app.get("/getmajornodecoords/")
async def fastapi_get_major_node_coords(network: str) -> list[str] | None:
coords = get_major_node_coords(network)
async def fastapi_get_major_node_coords(network: str, diameter: int) -> list[str] | None:
coords = get_major_node_coords(network, diameter)
result = []
for node_id, coord in coords.items():
result.append(f"{node_id}:{coord['type']}:{coord['x']}:{coord['y']}")
@@ -1423,8 +1423,8 @@ async def fastapi_get_network_link_nodes(network: str) -> list[str] | None:
# DingZQ 2024-12-31
# 获取直径大于800的管道
@app.get("/getmajorpipenodes/")
async def fastapi_get_major_pipe_nodes(network: str) -> list[str] | None:
return get_major_pipe_nodes(network)
async def fastapi_get_major_pipe_nodes(network: str, diameter: int) -> list[str] | None:
return get_major_pipe_nodes(network, diameter)
############################################################
# vertex 25.[VERTICES]