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: