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
|
# SPDX-License-Identifier: AGPL-3.0-or-later
import pygraphviz as pgv
def draw_setup(computers, networks, destination=None,
icon_prefix = '../../../static/icons/',
format='svg', icon_path = '', icon_suffix = None):
if icon_suffix is None:
icon_suffix = format
icon_suffix = '.' + icon_suffix
G = pgv.AGraph(imagepath=icon_path + '/')
print(G.graph_attr)
have_internet = []
for net in networks:
net_name = net.get('name', 'net')
if net.get('public', False):
have_internet.append(net_name)
G.add_node('net-' + net_name, label=net_name, shape='rectangle')
if len(have_internet):
G.add_node('net-' + 'internet',
label='internet',
labelloc='b',
image=icon_prefix + 'internet' + icon_suffix,
shape='none')
for n in have_internet:
G.add_edge('net-' + n, 'net-internet')
for properties in computers:
c = properties.get('name', '')
label_str = '< <TABLE BORDER="0"><TR><TD COLSPAN="2"><B>{}</B></TD></TR><TR><TD COLSPAN="2"><IMG SRC="{}"/></TD></TR>'
label = label_str.format(c, icon_prefix + 'computer' + icon_suffix)
for hdd in properties.get('disks', []):
icon = icon_prefix + 'drive-harddisk' + icon_suffix
label += '<TR><TD><IMG SRC="{}" /></TD><TD>{}</TD></TR>'.format(icon, hdd['name'])
label += '</TABLE> >'
G.add_node('comp-' + c,
label = label,
shape='box', labelloc='b')
for iface in properties.get('network_interfaces', []):
G.add_edge('comp-' + c, 'net-' + iface['network'])
return G.draw(path=destination, format=format, prog='dot')
if __name__ == '__main__':
import sample_task as task
print(draw_setup(task.computers, task.networks))
|
