tweaked module-dep graph tool

[sfa.git] / tools / py2depgraph.py

diff --git a/tools/py2depgraph.py b/tools/py2depgraph.py
index 15f60e8..022add3 100755 (executable)
--- 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: