diff --git a/api/s7_valves.py b/api/s7_valves.py index 4a24b1a..b9cbe22 100644 --- a/api/s7_valves.py +++ b/api/s7_valves.py @@ -120,3 +120,48 @@ def delete_valve_cache(name: str, cs: ChangeSet) -> SqlChangeSet: def delete_valve(name: str, cs: ChangeSet) -> ChangeSet: return execute_command(name, delete_valve_cache(name, cs)) + + +class InpValve: + 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.diameter = float(tokens[3]) + self.v_type = str(tokens[4]) + self.setting = float(tokens[5]) + self.minor_loss = float(tokens[6]) + self.desc = str(tokens[-1]) if has_desc else None + + +def inp_in_valve(section: list[str]) -> ChangeSet: + cs = ChangeSet() + for s in section: + # skip comment + if s.startswith(';'): + continue + obj = InpValve(s) + cs.append({'operation': API_ADD, 'type': 'valve', 'id': obj.id, 'node1': obj.node1, 'node2': obj.node2, 'diameter': obj.diameter, 'v_type': obj.v_type, 'setting': obj.setting, 'minor_loss': obj.minor_loss}) + return cs + + +def inp_out_valve(name: str) -> list[str]: + lines = [] + objs = read_all(name, 'select * from valves') + for obj in objs: + id = obj['id'] + node1 = obj['node1'] + node2 = obj['node2'] + diameter = obj['diameter'] + v_type = obj['type'] + setting = obj['setting'] + minor_loss = obj['minor_loss'] + desc = ';' + lines.append(f'{id} {node1} {node2} {diameter} {v_type} {setting} {minor_loss} {desc}') + return lines