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

Change major diameter to 0

Browse Source
This commit is contained in:
DingZQ
2025-01-01 00:43:56 +08:00
parent 212c9e7182
commit 51e3ae18bb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/s0_base.py
View File

@@ -111,7 +111,7 @@ def get_nodes_id_and_type(name: str) -> dict[str, str]:
def get_major_nodes(name: str) -> list[str]: def get_major_nodes(name: str) -> list[str]:
major_nodes_set = set() major_nodes_set = set()
with conn[name].cursor(row_factory=dict_row) as cur: with conn[name].cursor(row_factory=dict_row) as cur:
cur.execute(f"select node1, node2 from pipes where diameter > 100") cur.execute(f"select node1, node2 from pipes where diameter > 0")
for record in cur: for record in cur:
major_nodes_set.add(record['node1']) major_nodes_set.add(record['node1'])
major_nodes_set.add(record['node2']) major_nodes_set.add(record['node2'])
@@ -157,7 +157,7 @@ def get_links_id_and_type(name: str) -> dict[str, str]:
def get_major_pipes(name: str) -> list[str]: def get_major_pipes(name: str) -> list[str]:
major_pipe_ids: list[str] = [] major_pipe_ids: list[str] = []
with conn[name].cursor(row_factory=dict_row) as cur: with conn[name].cursor(row_factory=dict_row) as cur:
cur.execute(f"select id from pipes where diameter > 100 order by id") cur.execute(f"select id from pipes where diameter > 0 order by id")
for record in cur: for record in cur:
major_pipe_ids.append(record['id']) major_pipe_ids.append(record['id'])
return major_pipe_ids return major_pipe_ids