Add api addjunction
This commit is contained in:
8
main.py
8
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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user