Parse [RESERVOIRS]

api/s3_reservoirs.py

@@ -114,3 +114,37 @@ def delete_reservoir_cache(name: str, cs: ChangeSet) -> SqlChangeSet:
 
 def delete_reservoir(name: str, cs: ChangeSet) -> ChangeSet:
     return execute_command(name, delete_reservoir_cache(name, cs))
+
+
+class InpReservoir:
+    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.head = float(tokens[1])
+        self.pattern = float(tokens[2]) if num_without_desc >= 3 else None
+        self.desc = str(tokens[-1]) if has_desc else None
+
+
+def inp_in_reservoir(section: list[str]) -> ChangeSet:
+    cs = ChangeSet()
+    for s in section:
+        obj = InpReservoir(s)
+        cs.append({'operation': API_ADD, 'type': 'reservoir', 'id': obj.id, 'head': obj.head, 'pattern': obj.pattern})
+    return cs
+
+
+def inp_out_junction(name: str) -> list[str]:
+    lines = []
+    objs = read_all(name, 'select * from junctions')
+    for obj in objs:
+        id = obj['id']
+        head = obj['head']
+        pattern = obj['pattern'] if obj['pattern'] != None else ''
+        desc = ';'
+        lines.append(f'{id} {head} {pattern} {desc}')
+    return lines