Guard pump

api/s6_pumps.py

@@ -13,7 +13,9 @@ def get_pump_schema(name: str) -> dict[str, dict[str, Any]]:
 
 
 def get_pump(name: str, id: str) -> dict[str, Any]:
-    p = read(name, f"select * from pumps where id = '{id}'")
+    p = try_read(name, f"select * from pumps where id = '{id}'")
+    if p == None:
+        return {}
     d = {}
     d['id'] = str(p['id'])
     d['node1'] = str(p['node1'])
@@ -73,6 +75,10 @@ def set_pump_cmd(name: str, cs: ChangeSet) -> DbChangeSet:
 
 
 def set_pump(name: str, cs: ChangeSet) -> ChangeSet:
+    if 'id' not in cs.operations[0]:
+        return ChangeSet()
+    if get_pump(name, cs.operations[0]['id']) == {}:
+        return ChangeSet()
     return execute_command(name, set_pump_cmd(name, cs))
 
 
@@ -92,6 +98,10 @@ def add_pump_cmd(name: str, cs: ChangeSet) -> DbChangeSet:
 
 
 def add_pump(name: str, cs: ChangeSet) -> ChangeSet:
+    if 'id' not in cs.operations[0]:
+        return ChangeSet()
+    if get_pump(name, cs.operations[0]['id']) != {}:
+        return ChangeSet()
     return execute_command(name, add_pump_cmd(name, cs))
 
 
@@ -111,6 +121,10 @@ def delete_pump_cmd(name: str, cs: ChangeSet) -> DbChangeSet:
 
 
 def delete_pump(name: str, cs: ChangeSet) -> ChangeSet:
+    if 'id' not in cs.operations[0]:
+        return ChangeSet()
+    if get_pump(name, cs.operations[0]['id']) == {}:
+        return ChangeSet()
     return execute_command(name, delete_pump_cmd(name, cs))