Support inp in link new

This commit is contained in:
WQY\qiong
2023-03-15 22:04:39 +08:00
parent 81c4e0308b
commit 617c66d0cd
6 changed files with 77 additions and 4 deletions

View File

@@ -157,6 +157,26 @@ def inp_in_valve(section: list[str]) -> ChangeSet:
return cs
def inp_in_valve_new(name: str, line: str) -> None:
tokens = line.split()
num = len(tokens)
has_desc = tokens[-1].startswith(';')
num_without_desc = (num - 1) if has_desc else num
id = str(tokens[0])
node1 = str(tokens[1])
node2 = str(tokens[2])
diameter = float(tokens[3])
v_type = str(tokens[4].upper())
setting = str(tokens[5])
minor_loss = float(tokens[6])
desc = str(tokens[-1]) if has_desc else None
write(name, f"insert into _link (id, type) values ('{id}', 'valve');")
write(name, f"insert into valves (id, node1, node2, diameter, type, setting, minor_loss) values ('{id}', '{node1}', '{node2}', {diameter}, '{v_type}', '{setting}', {minor_loss});")
def inp_out_valve(name: str) -> list[str]:
lines = []
objs = read_all(name, 'select * from valves')