From 3250d5873330d05dfc33b44dfcff8bc11c8ce820 Mon Sep 17 00:00:00 2001
From: wqy <wqyuyu@outlook.com>
Date: Sat, 17 Sep 2022 11:10:28 +0800
Subject: [PATCH] Add pump api

---
 api/s6_pumps.py | 79 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100644 api/s6_pumps.py

diff --git a/api/s6_pumps.py b/api/s6_pumps.py
new file mode 100644
index 0000000..875265e
--- /dev/null
+++ b/api/s6_pumps.py
@@ -0,0 +1,79 @@
+from psycopg.rows import Row
+from .s0_base import *
+from .change_set import ChangeSet
+import utility
+
+
+def add_pump(name: str, id: str, node1: str, node2: str) -> ChangeSet:
+    if not is_node(name, node1):
+        return
+    if not is_node(name, node2):
+        return
+
+    sql = f"insert into pumps (id, node1, node2) values ('{id}', '{node1}', '{node2}'');"
+    undo_sql = f'delete from pumps where id = "{id}";'
+    return add_link(name, PUMP, id, sql, undo_sql)
+
+
+def _get_pump(name: str, id: str) -> Row | None:
+    return utility.query(name, f"select node1, node2 from pumps where id = '{id}'")
+
+
+def delete_pump(name: str, id: str) -> ChangeSet:
+    if not is_pump(name, id):
+        return
+
+    row = _get_pump(name, id)
+    if row == None:
+        return
+
+    node1, node2 = row['node1'], row['node2']
+
+    sql = f"delete from pumps where id = '{id}';"
+    undo_sql = f'insert into pumps (id, node1, node2) values ("{id}", "{node1}", "{node2}");'
+
+    return delete_link(name, PUMP, id, sql, undo_sql)
+
+
+def get_pump_node1(name: str, id: str) -> str | None:
+    row = _get_pump(name, id)
+    return float(row['node1']) if row != None else None
+
+
+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
+
+    row = _get_pump(name, id)
+    if row == None:
+        return
+
+    return utility.update(name, PUMP, 'pumps', 'id', id, key, key_type, row[key], value, optional)
+
+
+def set_pump_node1(name: str, id: str, node1: str) -> ChangeSet:
+    if not is_node(name, node1):
+        return
+
+    return _set_pump(name, id, 'node1', 'str', str(node1))
+
+
+def set_pump_node2(name: str, id: str, node2: str) -> ChangeSet:
+    if not is_node(name, node2):
+        return
+
+    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
\ No newline at end of file