From 51e3ae18bb5e2e0f1a66766ae68e4f1c889d5334 Mon Sep 17 00:00:00 2001 From: DingZQ Date: Wed, 1 Jan 2025 00:43:56 +0800 Subject: [PATCH] Change major diameter to 0 --- api/s0_base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/s0_base.py b/api/s0_base.py index 7a97cc1..f813d66 100644 --- a/api/s0_base.py +++ b/api/s0_base.py @@ -111,7 +111,7 @@ def get_nodes_id_and_type(name: str) -> dict[str, str]: def get_major_nodes(name: str) -> list[str]: major_nodes_set = set() 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: major_nodes_set.add(record['node1']) 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]: major_pipe_ids: list[str] = [] 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: major_pipe_ids.append(record['id']) return major_pipe_ids