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

Add more interface to base

Browse Source
This commit is contained in:
WQY\qiong
2023-04-30 11:28:40 +08:00
parent 3c233f6e9c
commit 5376ead768
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/s0_base.py
View File

@@ -1,5 +1,6 @@
from psycopg.rows import dict_row, Row from psycopg.rows import dict_row, Row
from .connection import g_conn_dict as conn from .connection import g_conn_dict as conn
from .database import read
_NODE = '_node' _NODE = '_node'
@@ -105,3 +106,14 @@ def get_node_links(name: str, id: str) -> list[str]:
for p in cur.execute(f"select id from valves where node1 = '{id}' or node2 = '{id}'").fetchall(): for p in cur.execute(f"select id from valves where node1 = '{id}' or node2 = '{id}'").fetchall():
links.append(p['id']) links.append(p['id'])
return links return links
def get_link_nodes(name: str, id: str) -> list[str]:
row = {}
if is_pipe(name, id):
row = read(name, f"select node1, node2 from pipes where id = '{id}'")
elif is_pump(name, id):
row = read(name, f"select node1, node2 from pumps where id = '{id}'")
elif is_valve(name, id):
row = read(name, f"select node1, node2 from valves where id = '{id}'")
return [str(row['node1']), str(row['node2'])]