分析并输出Python代码依赖的库代码,Python代码库,德仔网

代码语言

代码分类

【Python】分析并输出Python代码依赖的库

作者:DTC2 / 发布于2015/8/10/ 572

用法: 分析一个脚本的依赖: analysis_dependency.py script1.py 递归分析依赖: analysis_dependency.py script1.py -r


	#!/usr/bin/env python
	# encoding: utf-8
	# source: https://github.com/MrLYC/ycyc/blob/dev/tools/analysis_dependency.py
	
	import ast
	import importlib
	import inspect
	
	
	class Analysis(ast.NodeTransformer):
	 def __init__(self, paths, recursion):
	 self.modules = list()
	 self.paths = list(paths)
	 self.recursion = recursion
	
	 def add_module(self, module):
	 if module and module not in self.modules:
	 self.modules.append(module)
	 if self.recursion:
	 try:
	 path = inspect.getsourcefile(importlib.import_module(module))
	 if path:
	 self.paths.append(path)
	 except:
	 pass
	
	 def visit_Import(self, node):
	 for i in node.names:
	 self.add_module(i.name)
	
	 def visit_ImportFrom(self, node):
	 self.add_module(node.module)
	
	 def analysis(self):
	 for p in self.paths:
	 try:
	 with open(p, "rt") as fp:
	 self.visit(ast.parse(fp.read(), p))
	 except:
	 pass
	 return tuple(self.modules)
	
	if __name__ == "__main__":
	 import argparse
	
	 parser = argparse.ArgumentParser()
	 parser.add_argument("paths", nargs="+")
	 parser.add_argument("-r", "--recursion", action="store_true", default=False)
	 args = parser.parse_args()
	
	 analysisor = Analysis(args.paths, args.recursion)
	 for m in analysisor.analysis():
	 print m

评论列表

本站所提供的代码，版权归原作者所有，若有侵犯作者版权，请与我们联系，我们将立即删除或修改。谢谢!
本站所有代码发布及提供者。

试试其它关键字

　依赖的库　

同语言下

可能有用的

DTC2贡献的其它代码(12)