This commit is contained in:
xinzish
2024-04-08 23:41:36 +08:00
parent c1a5759220
commit 46f898a000
5 changed files with 33 additions and 7 deletions

View File

@@ -7,6 +7,7 @@ _NODE = '_node'
_LINK = '_link'
_CURVE = '_curve'
_PATTERN = '_pattern'
_REGION = '_region'
JUNCTION = 'junction'
RESERVOIR = 'reservoir'
@@ -18,6 +19,8 @@ VALVE = 'valve'
PATTERN = 'pattern'
CURVE = 'curve'
REGION = 'region'
def _get_from(name: str, id: str, base_type: str) -> Row | None:
with conn[name].cursor(row_factory=dict_row) as cur:
@@ -70,6 +73,9 @@ def is_curve(name: str, id: str) -> bool:
def is_pattern(name: str, id: str) -> bool:
return _get_from(name, id, _PATTERN) != None
def is_region(name: str, id: str) -> bool:
return _get_from(name, id, _REGION) != None
def _get_all(name: str, base_type: str) -> list[str]:
ids : list[str] = []
@@ -95,6 +101,8 @@ def get_curves(name: str) -> list[str]:
def get_patterns(name: str) -> list[str]:
return _get_all(name, _PATTERN)
def get_regions(name: str) -> list[str]:
return _get_all(name, _REGION)
def get_node_links(name: str, id: str) -> list[str]:
with conn[name].cursor(row_factory=dict_row) as cur: