1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
#!/usr/bin/python3
import xml.etree.ElementTree as ET
import sys
READ_OFF = 0
READ_FILE_PATH = 1
READ_OLD_NAME = 2
READ_NEW_NAME = 3
READ_BIG_VAR = 4
def get_new_port_name(port:str, old_group:str, new_group:str)->str:
if port.startswith((old_group + ':', old_group + '/')):
return port.replace(old_group, new_group, 1)
if port.startswith((old_group + '.0/', old_group + '.0:')):
return port.replace(old_group + '.0', new_group, 1)
return ''
def main_process():
input_file_path = ''
read_mode = READ_FILE_PATH
all_groups = []
new_group = {'old_name': '', 'new_name': ''}
for arg in sys.argv[1:]:
if read_mode == READ_FILE_PATH:
input_file_path = arg
read_mode = READ_OFF
elif arg == '-g':
read_mode = READ_OLD_NAME
elif read_mode == READ_OLD_NAME:
new_group['old_name'] = arg
read_mode = READ_NEW_NAME
elif read_mode == READ_NEW_NAME:
new_group['new_name'] = arg
all_groups.append(new_group)
new_group = {'old_name': '', 'new_name': ''}
read_mode = READ_OFF
elif arg.startswith('old_name:'):
read_mode = READ_BIG_VAR
for line in arg.splitlines():
if line.startswith('old_name:'):
new_group['old_name'] = line.replace('old_name:', '', 1)
elif line.startswith('new_name:'):
new_group['new_name'] = line.replace('new_name:', '', 1)
all_groups.append(new_group)
new_group = {'old_name': '', 'new_name': ''}
read_mode = READ_OFF
if not input_file_path:
sys.stderr.write('no input file, nothing to do.\n')
sys.exit(1)
if read_mode != READ_OFF:
sys.stderr.write('malformed arguments\n')
sys.exit(1)
try:
tree = ET.parse(input_file_path)
except:
sys.stderr.write('fail to parse %s as a XML file\n' % input_file_path)
sys.exit(1)
jackpatch_lines = []
new_connections = []
root = tree.getroot()
if root.tag != "RAY-JACKPATCH":
sys.stderr.write('xml file %s is not a RAY-JACKPATCH file\n' % input_file_path)
sys.exit(1)
for child in root:
if child.tag != "connection":
continue
port_from = ''
port_to = ''
for key in child.attrib.keys():
if key == 'from':
port_from = child.attrib[key]
elif key == 'to':
port_to = child.attrib[key]
if port_from and port_to:
break
else:
continue
jackpatch_lines.append("%s |> %s" % (port_from, port_to))
new_port_from = ''
new_port_to = ''
for group in all_groups:
# get_new_port_name returns an empty string
# if port may not belong to old group
if not new_port_from:
new_port_from = get_new_port_name(
port_from, group['old_name'], group['new_name'])
if not new_port_to:
new_port_to = get_new_port_name(
port_to, group['old_name'], group['new_name'])
if new_port_from and new_port_to:
break
if new_port_from:
if new_port_to:
jackpatch_lines.append("%s |> %s" % (new_port_from, new_port_to))
new_connections.append({'from': new_port_from, 'to': new_port_to})
jackpatch_lines.append("%s |> %s" % (new_port_from, port_to))
new_connections.append({'from': new_port_from, 'to': port_to})
if new_port_to:
jackpatch_lines.append("%s |> %s" % (port_from, new_port_to))
new_connections.append({'from': port_from, 'to': new_port_to})
for con in new_connections:
new_conn_xml = ET.SubElement(root, 'connection')
new_conn_xml.attrib['from'] = con['from']
new_conn_xml.attrib['to'] = con['to']
print('\n'.join(jackpatch_lines))
tree.write(input_file_path)
if __name__ == '__main__':
main_process()
|
