OrgTJWater/TJWaterServer
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add api addjunction

Browse Source
This commit is contained in:
DingZQ
2022-09-13 23:03:25 +08:00
parent 35f66d88f5
commit 846b3fc090
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@@ -8,7 +8,7 @@ from pydantic import BaseModel
from starlette.responses import FileResponse, JSONResponse from starlette.responses import FileResponse, JSONResponse
from fastapi import FastAPI, Response, status from fastapi import FastAPI, Response, status
import pytjnetwork as tj import pytjnetwork as tj
from tjnetwork import * from tjnetwork_new import *
JUNCTION = 0 JUNCTION = 0
RESERVOIR = 1 RESERVOIR = 1
@@ -83,6 +83,11 @@ async def fastapi_get_nodes(network: str):
return nodes 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/") @app.post("/addnode/")
async def fastapi_add_node(network: str, node: str): async def fastapi_add_node(network: str, node: str):
idx = add_node(network, node, JUNCTION) 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): async def fastapi_set_node_coord(network: str, node: str, x: float, y: float):
set_node_coord(network, node, x, y) set_node_coord(network, node, x, y)