simono41/docker-network-graph
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use fdp with splines and random network colours

Browse source
This commit is contained in:
LeoVerto 2018-08-31 06:21:48 +02:00
parent ad53549664
commit 98b38d11ac
2 changed files with 17 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
.gitignore vendored
View file

@ -1,3 +1,4 @@
*.gv *.gv
*.pdf *.pdf
*.png
.idea .idea

32
docker-net-graph.py
View file

@ -2,14 +2,13 @@
import os import os
import json import json
import argparse import argparse
import random
from docker import Client from docker import Client
from graphviz import Graph from graphviz import Graph
def generate_graph(verbose: bool, file: str): def generate_graph(verbose: bool, file: str):
dot = Graph(comment='Docker Network Graph', g = Graph(comment='Docker Network Graph', engine="fdp", format='png', graph_attr=dict(splines="true"))
graph_attr=dict(rankdir="TB", packmode='graph', pack='true')
)
docker_client = Client(os.environ.get("DOCKER_HOST", "unix:///var/run/docker.sock")) docker_client = Client(os.environ.get("DOCKER_HOST", "unix:///var/run/docker.sock"))
@ -32,14 +31,15 @@ def generate_graph(verbose: bool, file: str):
if verbose: if verbose:
print('|'.join(iface_labels)) print('|'.join(iface_labels))
dot.node(node_id, g.node(node_id,
shape='record', shape='record',
label="{ %s | { %s } }" % (name, '|'.join(iface_labels)), label="{ %s | { %s } }" % (name, '|'.join(iface_labels)),
fillcolor='#ff9999', fillcolor='#ff9999',
style='filled') style='filled')
for net in docker_client.networks(): for net in docker_client.networks():
net_name = net['Name'] net_name = net['Name']
color = "#%06x" % random.randint(0, 0xFFFFFF)
try: try:
gateway = net['IPAM']['Config'][0]['Gateway'] gateway = net['IPAM']['Config'][0]['Gateway']
@ -58,11 +58,11 @@ def generate_graph(verbose: bool, file: str):
net_label_html = '<br/>'.join([s for s in ['<font color="#777777"><i>network</i></font>', net_name, subnet, gateway] if s is not None]) net_label_html = '<br/>'.join([s for s in ['<font color="#777777"><i>network</i></font>', net_name, subnet, gateway] if s is not None])
dot.node(net_node_id, g.node(net_node_id,
shape='record', shape='record',
label="{<gw_iface> %s| %s }" % (gateway, net_name), label="{<gw_iface> %s| %s }" % (gateway, net_name),
fillcolor='#99ff99', fillcolor=color,
style='filled') style='filled')
for container_id, container in sorted(net['Containers'].items()): for container_id, container in sorted(net['Containers'].items()):
if verbose: if verbose:
@ -74,11 +74,11 @@ def generate_graph(verbose: bool, file: str):
container_iface_ref = "%s:%s" % (container_node_id, container['EndpointID']) container_iface_ref = "%s:%s" % (container_node_id, container['EndpointID'])
dot.edge(container_iface_ref, net_node_id+":gw_iface") g.edge(container_iface_ref, net_node_id+":gw_iface", color=color)
print(dot.source) print(g.source)
if file: if file:
dot.render(file) g.render(file)
if __name__ == "__main__": if __name__ == "__main__":