Add API to get all node properties

This commit is contained in:
DingZQ
2025-03-29 17:08:31 +08:00
parent d1687b8464
commit 0431f4d82e
6 changed files with 91 additions and 3 deletions

View File

@@ -26,6 +26,24 @@ def get_reservoir(name: str, id: str) -> dict[str, Any]:
d['links'] = get_node_links(name, id)
return d
# DingZQ, 2025-03-29
def get_all_reservoirs(name: str) -> list[dict[str, Any]]:
rows = read_all(name, f"select * from reservoirs")
if rows == None:
return []
result = []
for row in rows:
d = {}
d['id'] = str(row['id'])
d['x'] = float(row['x'])
d['y'] = float(row['y'])
d['head'] = float(row['head']) if row['head'] != None else None
d['pattern'] = str(row['pattern']) if row['pattern'] != None else None
d['links'] = get_node_links(name, row['id'])
result.append(d)
return result
class Reservoir(object):
def __init__(self, input: dict[str, Any]) -> None: