From 23733538227bc5b802694d234906543b354b5735 Mon Sep 17 00:00:00 2001 From: DingZQ Date: Fri, 21 Oct 2022 20:18:14 +0800 Subject: [PATCH] Add api deletenode and deletelink --- main.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/main.py b/main.py index 36381de..39ce537 100644 --- a/main.py +++ b/main.py @@ -778,6 +778,26 @@ async def fastapi_set_valve_properties(network: str, valve: str, req: Request) - return set_valve(network, ChangeSet(ps)) +# node & link +@app.post("/deletenode/") +async def fastapi_delete_node(network: str, node: str) -> ChangeSet: + ps = {'id' : node} + if is_junction(network, node): + return delete_junction(network, ChangeSet(ps)) + elif is_reservoir(network, node): + return delete_reservoir(network, ChangeSet(ps)) + elif is_tank(network, node): + return delete_tank(network, ChangeSet(ps)) + +@app.post("/deletelink/") +async def fastapi_delete_link(network: str, link: str) -> ChangeSet: + ps = {'id' : link} + if is_pipe(network, link): + return delete_pipe(network, ChangeSet(ps)) + elif is_pump(network, link): + return delete_pump(network, ChangeSet(ps)) + elif is_valve(network, link): + return delete_valve(network, ChangeSet(ps)) # inp file @app.post("/uploadinp/", status_code=status.HTTP_200_OK)