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)