Add pump api to tjnetwork
This commit is contained in:
@@ -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
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user