Parse [VALVES]
This commit is contained in:
@@ -120,3 +120,48 @@ def delete_valve_cache(name: str, cs: ChangeSet) -> SqlChangeSet:
|
|||||||
|
|
||||||
def delete_valve(name: str, cs: ChangeSet) -> ChangeSet:
|
def delete_valve(name: str, cs: ChangeSet) -> ChangeSet:
|
||||||
return execute_command(name, delete_valve_cache(name, cs))
|
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
|
||||||
|
|||||||
Reference in New Issue
Block a user