From 490674b5c63d3dd371e63eba2c45df07cc0fcde4 Mon Sep 17 00:00:00 2001 From: Thierry Parmentelat Date: Fri, 28 Oct 2011 09:42:31 +0200 Subject: [PATCH] tweaked module-dep graph tool --- tools/depgraph2dot.py | 4 ++-- tools/py2depgraph.py | 9 +++++++-- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/tools/depgraph2dot.py b/tools/depgraph2dot.py index 930e24f9..b8ecbce4 100755 --- a/tools/depgraph2dot.py +++ b/tools/depgraph2dot.py @@ -21,7 +21,7 @@ # SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. -import sys, getopt, colorsys, imp, md5 +import sys, getopt, colorsys, imp, hashlib class pydepgraphdot: @@ -179,7 +179,7 @@ class pydepgraphdot: return s[:i] def color_from_name(self,name): - n = md5.md5(name).digest() + n = hashlib.md5(name).digest() hf = float(ord(n[0])+ord(n[1])*0xff)/0xffff sf = float(ord(n[2]))/0xff vf = float(ord(n[3]))/0xff diff --git a/tools/py2depgraph.py b/tools/py2depgraph.py index 15f60e85..022add32 100755 --- a/tools/py2depgraph.py +++ b/tools/py2depgraph.py @@ -23,6 +23,8 @@ import sys, pprint import modulefinder +focus=[ 'sfa' , 'OpenSSL', 'M2Crypto', 'xmlrpclib', 'threading' ] + class mymf(modulefinder.ModuleFinder): def __init__(self,*args,**kwargs): self._depgraph = {} @@ -39,8 +41,11 @@ class mymf(modulefinder.ModuleFinder): self._last_caller = old_last_caller def import_module(self,partnam,fqname,parent): - if not fqname.startswith('sfa'): - print >>sys.stderr, "Trimmed fqname",fqname + keep=False + for start in focus: + if fqname.startswith(start): keep=True + if not keep: + print >> sys.stderr, "Trimmed fqname",fqname return r = modulefinder.ModuleFinder.import_module(self,partnam,fqname,parent) if r is not None: -- 2.43.0