跳到主要内容

analyze packages imported by python project

import os
import re

def analyze_imports(project_path):
imports = set()

for root, dirs, files in os.walk(project_path):
for file in files:
if file.endswith(".py"):
file_path = os.path.join(root, file)
with open(file_path, "r") as f:
lines = f.readlines()
for line in lines:
if line.startswith("import"):
match = re.findall(r"[\w.]+", line)
if match:
import_name = match[1].split('.')[0] # 只保留包名部分
imports.add(import_name)
elif line.startswith("from"):
match = re.findall(r"[\w.]+", line)
if match:
import_name = match[0].split('.')[0] # 只保留包名部分
imports.add(import_name)

return imports

project_path = "/path/to/your/project"
result = analyze_imports(project_path)

for import_name in result:
print(import_name)