Add api addjunction

This commit is contained in:
DingZQ
2022-09-13 23:03:25 +08:00
parent 35f66d88f5
commit 846b3fc090

View File

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