From 5901955fe70b5bb73c869d798854525f596b0def Mon Sep 17 00:00:00 2001
From: Mike Crute <mcrute@gmail.com>
Date: Sun, 26 Jul 2009 13:24:39 -0400
Subject: Re-adding the cleanup patch

---
 py2depgraph.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/py2depgraph.py b/py2depgraph.py
index bff29fd..f8318d1 100644
--- a/py2depgraph.py
+++ b/py2depgraph.py
@@ -49,6 +49,12 @@ class mymf(modulefinder.ModuleFinder):
             self._types[r.__name__] = type
         return r
 
+def clean_graph(graph, package):
+    for key, value in graph.items():
+        if not key.startswith(package):
+            del graph[key]
+
+    return graph
 
 def main(argv):
     path = sys.path[:]
@@ -56,7 +62,9 @@ def main(argv):
     exclude = []
     mf = mymf(path,debug,exclude)
     mf.run_script(argv[0])
-    pprint.pprint({'depgraph':mf._depgraph,'types':mf._types})
+    app_name = argv[1]
+    pprint.pprint({'depgraph':clean_graph(mf._depgraph, app_name),
+                   'types':clean_graph(mf._types, app_name)})
 
 if __name__=='__main__':
     main(sys.argv[1:])
-- 
cgit v1.2.3