使用Python AST更改导入路径的最佳方法

import ast

def change_import_path(code, old_path, new_path):
    # 解析源代码，生成AST
    tree = ast.parse(code)

    # 遍历AST，查找导入语句
    for node in ast.walk(tree):
        if isinstance(node, ast.Import):
            # 对于Import节点，检查导入路径是否符合修改条件
            for alias in node.names:
                if alias.name == old_path:
                    # 修改导入路径
                    alias.name = new_path
        elif isinstance(node, ast.ImportFrom):
            # 对于ImportFrom节点，检查导入路径是否符合修改条件
            if node.module == old_path:
                # 修改导入路径
                node.module = new_path

    # 将修改后的AST转换回源代码字符串
    new_code = ast.unparse(tree)

    return new_code

# 示例调用
code = '''
import os
from utils import helper
'''
old_path = 'utils'
new_path = 'new_utils'
new_code = change_import_path(code, old_path, new_path)
print(new_code)