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

@@ -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}";'