From e488686c3c1c7323ff1055c0ce5ab563219a7c39 Mon Sep 17 00:00:00 2001 From: DingZQ Date: Thu, 15 Sep 2022 23:54:45 +0800 Subject: [PATCH] Add node methods --- main.py | 30 +++++++++++++++++++++++++++++- 1 file changed, 29 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index c110ddb..552ec35 100644 --- a/main.py +++ b/main.py @@ -53,6 +53,10 @@ async def fastapi_create_project(network: str): print(network) return network +@app.get("/isprojectopen/") +async def fastapi_is_project_open(network: str): + return is_project_open(network) + @app.post("/openproject/") async def fastapi_open_project(network: str): open_project(network) @@ -70,11 +74,35 @@ async def fastapi_delete_project(network: str): delete_project(network) return True -# element operations +@app.post("/copyproject/") +async def fastapi_copy_project(source: str, target: str): + copy_project(source, target) + return True + +# node @app.get("/getnodes/") async def fastapi_get_nodes(network: str) -> list[str]: return get_nodes(network) +@app.get("/isnode/") +async def fastapi_is_node(network: str, node: str) -> bool: + return is_node(network, node) + +@app.get("/isjunction/") +async def fastapi_is_junction(network: str, node: str) -> bool: + return is_junction(network, node) + +@app.get("/isreservoir/") +async def fastapi_is_reservoir(network: str, node: str) -> bool: + return is_reservoir(network, node) + +@app.get("/istank/") +async def fastapi_is_tank(network: str, node: str) -> bool: + return is_tank(network, node) + + + + # junction @app.post("/addjunction/") async def fastapi_add_junction(network: str, junction: str, x: float, y: float, z: float) -> ChangeSet: