Add api deletenode and deletelink

This commit is contained in:
DingZQ
2022-10-21 20:18:14 +08:00
parent 6a66f8c967
commit 2373353822

20
main.py
View File

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