From 9905d7f02240166316c46355ee74e0755f76523e Mon Sep 17 00:00:00 2001 From: wqy Date: Thu, 22 Sep 2022 21:45:25 +0800 Subject: [PATCH] Include links property --- api/s3_reservoirs.py | 2 +- api/s4_tanks.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/api/s3_reservoirs.py b/api/s3_reservoirs.py index 35ec8de..a8c5430 100644 --- a/api/s3_reservoirs.py +++ b/api/s3_reservoirs.py @@ -63,7 +63,7 @@ def set_reservoir_coord(name: str, id: str, x: float, y: float) -> ChangeSet: def get_reservoir_property_names(name: str) -> list[str]: - return ['head', 'pattern', 'coord'] + return ['head', 'pattern', 'coord', 'links'] def get_reservoir_properties(name: str, id: str) -> dict[str, Any] | None: diff --git a/api/s4_tanks.py b/api/s4_tanks.py index 59f15f1..ae3d37d 100644 --- a/api/s4_tanks.py +++ b/api/s4_tanks.py @@ -100,7 +100,7 @@ def set_tank_coord(name: str, id: str, x: float, y: float) -> ChangeSet: def get_tank_property_names(name: str) -> list[str]: - return ['elevation', 'init_level', 'min_level', 'max_level', 'diameter', 'min_vol', 'vol_curve', 'overflow', 'coord'] + return ['elevation', 'init_level', 'min_level', 'max_level', 'diameter', 'min_vol', 'vol_curve', 'overflow', 'coord', 'links'] def get_tank_properties(name: str, id: str) -> dict[str, Any] | None: @@ -118,4 +118,5 @@ def get_tank_properties(name: str, id: str) -> dict[str, Any] | None: ps['vol_curve'] = row['vol_curve'] if row != None and row['vol_curve'] != None else None ps['overflow'] = row['overflow'] if row != None and row['overflow'] != None else None ps['coord'] = get_node_coord(name, id) + ps['links'] = get_node_links(name, id) return ps