diff options
-rw-r--r-- | py2depgraph.py | 10 |
1 files changed, 9 insertions, 1 deletions
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): | |||
49 | self._types[r.__name__] = type | 49 | self._types[r.__name__] = type |
50 | return r | 50 | return r |
51 | 51 | ||
52 | def clean_graph(graph, package): | ||
53 | for key, value in graph.items(): | ||
54 | if not key.startswith(package): | ||
55 | del graph[key] | ||
56 | |||
57 | return graph | ||
52 | 58 | ||
53 | def main(argv): | 59 | def main(argv): |
54 | path = sys.path[:] | 60 | path = sys.path[:] |
@@ -56,7 +62,9 @@ def main(argv): | |||
56 | exclude = [] | 62 | exclude = [] |
57 | mf = mymf(path,debug,exclude) | 63 | mf = mymf(path,debug,exclude) |
58 | mf.run_script(argv[0]) | 64 | mf.run_script(argv[0]) |
59 | pprint.pprint({'depgraph':mf._depgraph,'types':mf._types}) | 65 | app_name = argv[1] |
66 | pprint.pprint({'depgraph':clean_graph(mf._depgraph, app_name), | ||
67 | 'types':clean_graph(mf._types, app_name)}) | ||
60 | 68 | ||
61 | if __name__=='__main__': | 69 | if __name__=='__main__': |
62 | main(sys.argv[1:]) | 70 | main(sys.argv[1:]) |