Add more checking for link api

This commit is contained in:
wqy
2022-09-17 22:26:03 +08:00
parent 5dd83d7652
commit 939cef4f1e
3 changed files with 9 additions and 0 deletions

View File

@@ -14,6 +14,11 @@ def add_pipe(name: str, id: str, node1: str, node2: str, length: float = 0, diam
return ChangeSet()
if not is_node(name, node2):
return ChangeSet()
if node1 == node2:
return ChangeSet()
if status != PIPE_STATUS_OPEN and status != PIPE_STATUS_CLOSED and status != PIPE_STATUS_CV:
return ChangeSet()
sql = f"insert into pipes (id, node1, node2, length, diameter, roughness, minor_loss, status) values ('{id}', '{node1}', '{node2}', {length}, {diameter}, {roughness}, {minor_loss}, '{status}');"
undo_sql = f'delete from pipes where id = "{id}";'

View File

@@ -9,6 +9,8 @@ def add_pump(name: str, id: str, node1: str, node2: str) -> ChangeSet:
return ChangeSet()
if not is_node(name, node2):
return ChangeSet()
if node1 == node2:
return ChangeSet()
sql = f"insert into pumps (id, node1, node2) values ('{id}', '{node1}', '{node2}'');"
undo_sql = f'delete from pumps where id = "{id}";'

View File

@@ -17,6 +17,8 @@ def add_valve(name: str, id: str, node1: str, node2: str, diameter: float = 0, t
return ChangeSet()
if not is_node(name, node2):
return ChangeSet()
if node1 == node2:
return ChangeSet()
sql = f"insert into valves (id, node1, node2, diameter, type, setting, minor_loss) values ('{id}', '{node1}', '{node2}', {diameter}, '{type}', {setting}, {minor_loss});"
undo_sql = f'delete from valves where id = "{id}";'