From cef4e90704eae3fb4903554d3f8cb102b73b03a7 Mon Sep 17 00:00:00 2001 From: DingZQ Date: Thu, 22 Sep 2022 22:33:27 +0800 Subject: [PATCH] Support reservoir coord method --- main.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index f231825..073df9a 100644 --- a/main.py +++ b/main.py @@ -203,7 +203,8 @@ async def fastapi_delete_reservoir(network: str, reservoir: str) -> ChangeSet: @app.get("/getreservoirhead/") async def fastapi_get_reservoir_head(network: str, reservoir: str) -> float | None: - return get_reservoir_head(network, reservoir) + ps = get_reservoir_properties(network, reservoir) + return ps['head'] @app.get("/getreservoirpattern/") async def fastapi_get_reservoir_pattern(network: str, reservoir: str) -> str | None: @@ -211,7 +212,8 @@ async def fastapi_get_reservoir_pattern(network: str, reservoir: str) -> str | N @app.get("/getreservoircoord/") async def fastapi_get_reservoir_coord(network: str, reservoir: str) -> dict[str, float] | None: - return get_reservoir_coord(network, reservoir) + ps = get_reservoir_properties(network, reservoir) + return ps['coord'] @app.post("/setreservoirhead/") async def fastapi_set_reservoir_head(network: str, reservoir: str, head: float) -> ChangeSet: