modgraph.py

来自「Graphviz - Graph Drawing Programs from A」· Python 代码 · 共 39 行

#!/usr/bin/python

# display the kernel module dependencies

# author: Michael Hohn <[e-mail]>
#  based on: modgraph.tcl by John Ellson <[e-mail]>

import sys
sys.path.append('/usr/lib64/graphviz/python')
sys.path.append('/usr/lib/graphviz/python')
import gv

modules = open("/proc/modules", 'r').readlines()

G = gv.digraph("G")
N = gv.protonode(G)
E = gv.protoedge(G)

gv.setv(G, 'rankdir', 'LR')
gv.setv(G, 'nodesep', '0.05')
gv.setv(N, 'shape', 'box')
gv.setv(N, 'width', '0')
gv.setv(N, 'height', '0')
gv.setv(N, 'margin', '.03')
gv.setv(N, 'fontsize', '8')
gv.setv(N, 'fontname', 'helvetica')
gv.setv(E, 'arrowsize', '.4')

for rec in modules:
   fields = rec.split(' ')
   n = gv.node(G, fields[0])
   for usedby in fields[3].split(','):
      if (usedby != '-') & (usedby != ''):
         gv.edge(n, gv.node(G, usedby))

gv.layout(G, 'dot')
# The 'gtk' renderer is provided by graphviz-cairo
gv.render(G, 'png', 'modgraph.png')