From 5ab7246317f799b8ccb8b1c0033bc2273ea37a1e Mon Sep 17 00:00:00 2001 From: DingZQ Date: Sun, 22 Jun 2025 23:08:47 +0800 Subject: [PATCH] Add parameter token for api fastapi_get_network_geometries --- main.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index b484447..f40261f 100644 --- a/main.py +++ b/main.py @@ -1772,7 +1772,10 @@ async def fastapi_get_node_coord(network: str, node: str) -> dict[str, float] | # node type: junction, reservoir, tank # link type: pipe, pump, valve @app.get("/getnetworkgeometries/") -async def fastapi_get_network_geometries(network: str) -> dict[str, Any] | None: +async def fastapi_get_network_geometries(network: str, token: str) -> dict[str, Any] | None: + if token != "567e33c876a2": + raise HTTPException(status_code=401, detail="Unauthorized") + # 获取所有节点坐标# 缓存查询结果提高性能 global redis_client cache_key = f"getnetworkgeometries_{network}"