Add pump api to tjnetwork

This commit is contained in:
wqy
2022-09-17 11:16:45 +08:00
parent 3250d58733
commit 4350b5d19e
3 changed files with 66 additions and 35 deletions

View File

@@ -36,4 +36,9 @@ from .s5_pipes import add_pipe, delete_pipe
from .s5_pipes import get_pipe_node1, get_pipe_node2, get_pipe_length, get_pipe_diameter, get_pipe_roughness, get_pipe_minor_loss, get_pipe_status
from .s5_pipes import set_pipe_node1, set_pipe_node2, set_pipe_length, set_pipe_diameter, set_pipe_roughness, set_pipe_minor_loss, set_pipe_status
from .s6_pumps import add_pump, delete_pump
from .s6_pumps import get_pump_node1, get_pump_node2
from .s6_pumps import set_pump_node1, set_pump_node2
# TODO: more properties...
from .s24_coordinates import get_node_coord, set_node_coord

View File

@@ -44,11 +44,13 @@ def get_pump_node2(name: str, id: str) -> str | None:
row = _get_pump(name, id)
return float(row['node2']) if row != None else None
def get_pump_power(name: str, id: str) -> str | None: pass
def get_pump_speed(name: str, id: str) -> str | None: pass
def get_pump_head(name: str, id: str) -> str | None: pass
def get_pump_pattern(name: str, id: str) -> str | None: pass
def _set_pump(name: str, id: str, key: str, key_type: str, value: str, optional: bool = False) -> ChangeSet:
if not is_pump(name, id):
return
@@ -73,7 +75,8 @@ def set_pump_node2(name: str, id: str, node2: str) -> ChangeSet:
return _set_pump(name, id, 'node2', 'str', str(node2))
def set_pump_power(name: str, id: str) -> ChangeSet: pass
def set_pump_speed(name: str, id: str) -> ChangeSet: pass
def set_pump_head(name: str, id: str) -> ChangeSet: pass
def set_pump_pattern(name: str, id: str) -> ChangeSet: pass
def set_pump_pattern(name: str, id: str) -> ChangeSet: pass