diff options
Diffstat (limited to 'py2depgraph.py')
-rw-r--r-- | py2depgraph.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/py2depgraph.py b/py2depgraph.py index 5e97e06..bff29fd 100644 --- a/py2depgraph.py +++ b/py2depgraph.py | |||
@@ -28,7 +28,7 @@ class mymf(modulefinder.ModuleFinder): | |||
28 | self._types = {} | 28 | self._types = {} |
29 | self._last_caller = None | 29 | self._last_caller = None |
30 | modulefinder.ModuleFinder.__init__(self,*args,**kwargs) | 30 | modulefinder.ModuleFinder.__init__(self,*args,**kwargs) |
31 | 31 | ||
32 | def import_hook(self, name, caller=None, fromlist=None): | 32 | def import_hook(self, name, caller=None, fromlist=None): |
33 | old_last_caller = self._last_caller | 33 | old_last_caller = self._last_caller |
34 | try: | 34 | try: |
@@ -36,27 +36,27 @@ class mymf(modulefinder.ModuleFinder): | |||
36 | return modulefinder.ModuleFinder.import_hook(self,name,caller,fromlist) | 36 | return modulefinder.ModuleFinder.import_hook(self,name,caller,fromlist) |
37 | finally: | 37 | finally: |
38 | self._last_caller = old_last_caller | 38 | self._last_caller = old_last_caller |
39 | 39 | ||
40 | def import_module(self,partnam,fqname,parent): | 40 | def import_module(self,partnam,fqname,parent): |
41 | r = modulefinder.ModuleFinder.import_module(self,partnam,fqname,parent) | 41 | r = modulefinder.ModuleFinder.import_module(self,partnam,fqname,parent) |
42 | if r is not None: | 42 | if r is not None: |
43 | self._depgraph.setdefault(self._last_caller.__name__,{})[r.__name__] = 1 | 43 | self._depgraph.setdefault(self._last_caller.__name__,{})[r.__name__] = 1 |
44 | return r | 44 | return r |
45 | 45 | ||
46 | def load_module(self, fqname, fp, pathname, (suffix, mode, type)): | 46 | def load_module(self, fqname, fp, pathname, (suffix, mode, type)): |
47 | r = modulefinder.ModuleFinder.load_module(self, fqname, fp, pathname, (suffix, mode, type)) | 47 | r = modulefinder.ModuleFinder.load_module(self, fqname, fp, pathname, (suffix, mode, type)) |
48 | if r is not None: | 48 | if r is not None: |
49 | self._types[r.__name__] = type | 49 | self._types[r.__name__] = type |
50 | return r | 50 | return r |
51 | 51 | ||
52 | 52 | ||
53 | def main(argv): | 53 | def main(argv): |
54 | path = sys.path[:] | 54 | path = sys.path[:] |
55 | debug = 0 | 55 | debug = 0 |
56 | exclude = [] | 56 | exclude = [] |
57 | mf = mymf(path,debug,exclude) | 57 | mf = mymf(path,debug,exclude) |
58 | mf.run_script(argv[0]) | 58 | mf.run_script(argv[0]) |
59 | pprint.pprint({'depgraph':mf._depgraph,'types':mf._types}) | 59 | pprint.pprint({'depgraph':mf._depgraph,'types':mf._types}) |
60 | 60 | ||
61 | if __name__=='__main__': | 61 | if __name__=='__main__': |
62 | main(sys.argv[1:]) \ No newline at end of file | 62 | main(sys.argv[1:]) |