diff options
Diffstat (limited to 'kpov_judge/web/kpov_judge/kpov_draw_setup.py')
| -rw-r--r-- | kpov_judge/web/kpov_judge/kpov_draw_setup.py | 43 |
1 files changed, 0 insertions, 43 deletions
diff --git a/kpov_judge/web/kpov_judge/kpov_draw_setup.py b/kpov_judge/web/kpov_judge/kpov_draw_setup.py deleted file mode 100644 index 41f5e01..0000000 --- a/kpov_judge/web/kpov_judge/kpov_draw_setup.py +++ /dev/null @@ -1,43 +0,0 @@ -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)) - |