Support reservoir coord method

This commit is contained in:
DingZQ
2022-09-22 22:33:27 +08:00
parent 9d01fabf27
commit cef4e90704

View File

@@ -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: