Python扫描目录提取import语句
Python脚本用于扫描指定目录下所有的.py文件，提取其中所有的import语句中的模块名（包括相对导入），去重后按时间戳命名输出到文本文件。
#!/usr/bin/env python3# -*- coding: utf-8 -*-""" 扫描目录下所有 Python 文件，提取所有 import 语句中的模块名， 去重后保存到以时间戳命名的文本文件中。 """importosimportsysimportastimportargparsefromdatetimeimportdatetimedefextract_imports_from_file(filepath):""" 解析单个 Python 文件，返回该文件中导入的所有模块名（集合）。 支持： - import module - import module as alias - from module import name - from module import name as alias - 相对导入（from . import x, from ..sub import y 等） """modules=set()try:withopen(filepath,'r',encoding='utf-8')asf:content=f.read()tree=ast.parse(content,filename=filepath)fornodeinast.walk(tree):ifisinstance(node,ast.Import):# import module, import module as aliasforaliasinnode.names:modules.add(alias.name)# 完整模块名elifisinstance(node,ast.ImportFrom):# from module import ...ifnode.module:# 绝对导入或相对导入但指定了模块（如 from .sub import x）ifnode.level>0:# 相对导入，模块名前加对应数量的点module='.'*node.level+node.moduleelse:module=node.moduleelse:# 纯相对导入（如 from . import x, from .. import y）module='.'*node.level modules.add(module)exceptExceptionase:print(f"警告：解析文件{filepath}时出错：{e}",file=sys.stderr)returnmodulesdefmain():parser=argparse.ArgumentParser(description='扫描目录下的 Python 文件，提取所有 import 模块名并去重输出。')parser.add_argument('directory',nargs='?',default='.',help='要扫描的目录（默认：当前目录）')args=parser.parse_args()target_dir=args.directoryifnotos.path.isdir(target_dir):print(f"错误：'{target_dir}' 不是一个有效的目录。",file=sys.stderr)sys.exit(1)all_modules=set()# 递归遍历目录forroot,dirs,filesinos.walk(target_dir):forfileinfiles:iffile.endswith('.py'):filepath=os.path.join(root,file)modules=extract_imports_from_file(filepath)all_modules.update(modules)ifnotall_modules:print("未找到任何 import 语句。")return# 排序后输出sorted_modules=sorted(all_modules)# 生成时间戳文件名timestamp=datetime.now().strftime("%Y%m%d_%H%M%S")output_filename=f"imports_{timestamp}.txt"withopen(output_filename,'w',encoding='utf-8')asf:formoduleinsorted_modules:f.write(module+'\n')print(f"共找到{len(sorted_modules)}个唯一模块。")print(f"结果已写入：{output_filename}")if__name__=='__main__':main()使用方法
- 将上述代码保存为
.py文件（例如scan_imports.py）。 - 在终端中运行：
如果不指定目录，则默认扫描当前目录。python scan_imports.py[要扫描的目录] - 脚本会在当前目录下生成一个类似
imports_20250311_143022.txt的文件，每行列出一个模块名（已去重）。
功能说明
- 递归扫描指定目录下所有
.py文件。 - 使用 Python 的
ast模块解析代码，准确提取各种形式的import语句（包括别名和相对导入）。 - 对于
from ... import ...，记录from后面的模块名（相对导入会保留前导点号）。 - 对于
import ...，记录import后面的模块名（包括点号分隔的子模块）。 - 去重后按字母顺序排序，输出到以时间戳命名的文本文件。