Return type from getnetworkcoord

This commit is contained in:
DingZQ
2024-12-18 22:41:35 +08:00
parent f8ac46d80d
commit 70afb19184
4 changed files with 21 additions and 7 deletions

View File

@@ -29,7 +29,7 @@ from .s0_base import is_node, is_junction, is_reservoir, is_tank
from .s0_base import is_link, is_pipe, is_pump, is_valve
from .s0_base import is_curve
from .s0_base import is_pattern
from .s0_base import get_nodes, get_links, get_curves, get_patterns
from .s0_base import get_nodes, get_junctions, get_reservoirs, get_tanks, get_links, get_pipes, get_pumps, get_valves, get_curves, get_patterns
from .s0_base import get_node_links, get_link_nodes
from .s1_title import get_title_schema, get_title, set_title

View File

@@ -103,18 +103,17 @@ def get_junctions(name: str) -> list[str]:
return _get_nodes_by_type(name, JUNCTION)
# DingZQ
def get_reservoir(name: str) -> list[str]:
def get_reservoirs(name: str) -> list[str]:
return _get_nodes_by_type(name, RESERVOIR)
# DingZQ
def get_tank(name: str) -> list[str]:
def get_tanks(name: str) -> list[str]:
return _get_nodes_by_type(name, TANK)
# DingZQ
def get_links(name: str) -> list[str]:
return _get_all(name, _LINK)
# DingZQ
def _get_links_by_type(name: str, type: str) -> list[str]:
ids : list[str] = []