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

Add api to query type

Browse Source
This commit is contained in:
wqy
2022-09-02 19:01:59 +08:00
parent d90f7d6f8e
commit 49bea820b8
3 changed files with 37 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
api/s0_base.py
View File

@@ -78,9 +78,27 @@ def _get_impl(name: str, id: str, table: str) -> dict[str, str]:
def get_node(name: str, id: str) -> dict[str, str]:
return _get_impl(name, id, _NODE)
def is_junction(name: str, id: str) -> bool:
return get_node(name, id)['type'] == JUNCTION
def is_reservoir(name: str, id: str) -> bool:
return get_node(name, id)['type'] == RESERVOIR
def is_tank(name: str, id: str) -> bool:
return get_node(name, id)['type'] == TANK
def get_link(name: str, id: str) -> dict[str, str]:
return _get_impl(name, id, _LINK)
def is_pipe(name: str, id: str) -> bool:
return get_link(name, id)['type'] == PIPE
def is_pump(name: str, id: str) -> bool:
return get_link(name, id)['type'] == PUMP
def is_valve(name: str, id: str) -> bool:
return get_link(name, id)['type'] == VALVE
def get_curve(name: str, id: str) -> dict[str, str]:
return _get_impl(name, id, _CURVE)