From 846b3fc0908b9843a70f1a51d7f462884d4d765a Mon Sep 17 00:00:00 2001 From: DingZQ Date: Tue, 13 Sep 2022 23:03:25 +0800 Subject: [PATCH] Add api addjunction --- main.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 86c5c44..b73e5d6 100644 --- a/main.py +++ b/main.py @@ -8,7 +8,7 @@ from pydantic import BaseModel from starlette.responses import FileResponse, JSONResponse from fastapi import FastAPI, Response, status import pytjnetwork as tj -from tjnetwork import * +from tjnetwork_new import * JUNCTION = 0 RESERVOIR = 1 @@ -83,6 +83,11 @@ async def fastapi_get_nodes(network: str): return nodes +# junction +@app.post("/addjunction/") +async def fastapi_add_junction(network: str, junction: str, x: float, y: float, z: float) -> ChangeSet: + return add_junction(network, junction, x, y, z) + @app.post("/addnode/") async def fastapi_add_node(network: str, node: str): idx = add_node(network, node, JUNCTION) @@ -128,7 +133,6 @@ async def fastapi_get_node_coord(network: str, node: str): } ) -@app.post("/setnodecoord/") async def fastapi_set_node_coord(network: str, node: str, x: float, y: float): set_node_coord(network, node, x, y)