From 8fa9fc5b9887b32bddd05a1c1f1273b793d5cdbe Mon Sep 17 00:00:00 2001
From: "WQY\\qiong" <qiongyu.wang@outlook.com>
Date: Sat, 12 Nov 2022 15:25:33 +0800
Subject: [PATCH] Parse [PUMPS]

---
 api/s6_pumps.py | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/api/s6_pumps.py b/api/s6_pumps.py
index 3fc4bb8..04781d2 100644
--- a/api/s6_pumps.py
+++ b/api/s6_pumps.py
@@ -112,3 +112,51 @@ def delete_pump_cache(name: str, cs: ChangeSet) -> SqlChangeSet:
 
 def delete_pump(name: str, cs: ChangeSet) -> ChangeSet:
     return execute_command(name, delete_pump_cache(name, cs))
+
+
+class InpPump:
+    def __init__(self, line: str) -> None:
+        tokens = line.split()
+
+        num = len(tokens)
+        has_desc = tokens[-1].startswith(';')
+        num_without_desc = (num - 1) if has_desc else num
+
+        self.id = str(tokens[0])
+        self.node1 = str(tokens[1])
+        self.node2 = str(tokens[2])
+        self.props = {}
+        for i in range(3, num_without_desc, 2):
+            self.props |= { tokens[i].lower(): tokens[i + 1] }
+        self.power = float(self.props['power']) if 'power' in self.props else None
+        self.head = str(self.props['head']) if 'head' in self.props else None
+        self.speed = float(self.props['speed']) if 'speed' in self.props else None
+        self.pattern = str(self.props['pattern']) if 'pattern' in self.props else None
+        self.desc = str(tokens[-1]) if has_desc else None
+
+
+def inp_in_pump(section: list[str]) -> ChangeSet:
+    cs = ChangeSet()
+    for s in section:
+        # skip comment
+        if s.startswith(';'):
+            continue
+        obj = InpPump(s)
+        cs.append({'operation': API_ADD, 'type': 'pump', 'id': obj.id, 'node1': obj.node1, 'node2': obj.node2, 'power': obj.power, 'head': obj.head, 'speed': obj.speed, 'pattern': obj.pattern})
+    return cs
+
+
+def inp_out_pump(name: str) -> list[str]:
+    lines = []
+    objs = read_all(name, 'select * from pumps')
+    for obj in objs:
+        id = obj['id']
+        node1 = obj['node1']
+        node2 = obj['node2']
+        power = f"POWER {obj['power']}" if obj['power'] != None else ''
+        head = f"HEAD {obj['head']}" if obj['head'] != None else ''
+        speed = f"SPEED {obj['speed']}" if obj['speed'] != None else ''
+        pattern = f"PATTERN {obj['pattern']}" if obj['pattern'] != None else ''
+        desc = ';'
+        lines.append(f'{id} {node1} {node2} {power} {head} {speed} {pattern} {desc}')
+    return lines