如何使用命令行工具批量自动更新Python代码中的import导入语句

因为导入语句存在别名、批量导入、换行等多种写法，纯正则容易出现误匹配/漏匹配，优先选择语法感知的工具完成修改，以下是3种可直接落地的方案：

方案1：LibCST自定义脚本（精准度最高，覆盖所有Python语法场景） #

LibCST是Python生态专门用于代码重构的语法解析库，能在保留代码格式、注释、空行的前提下修改语法树，完全避免正则的弊端。

• 安装依赖：pip install libcst
• 编写重构脚本import_rename.py：

import libcst as cst
from libcst.codemod import CodemodContext, VisitorBasedCodemodCommand
from libcst.codemod.visitors import AddImportsVisitor, RemoveImportsVisitor

class ImportRenameCommand(VisitorBasedCodemodCommand):
    DESCRIPTION = "Rename a.b.foo_method to b.d.bar_method"
    
    def visit_ImportFrom(self, node: cst.ImportFrom) -> None:
        # 匹配 from a.b import 相关语句
        if node.module and cst.ensure_type(node.module, cst.Name).value == "a.b":
            for alias in node.names:
                if alias.name.value == "foo_method":
                    # 移除旧导入，添加新导入
                    RemoveImportsVisitor.add_needed_import(self.context, "a.b", "foo_method")
                    AddImportsVisitor.add_needed_import(self.context, "b.d", "bar_method")
                    return None
        return node

    def visit_Name(self, node: cst.Name) -> None:
        # 匹配所有 foo_method 调用，替换为新方法名
        if node.value == "foo_method":
            return cst.Name(value="bar_method")
        return node

• 执行全目录重构：python -m libcst.codemod --codemod import_rename.py 你的源码根目录

这个方案能自动适配别名导入、批量导入、跨文件调用等所有合法Python语法场景，不会误改注释、字符串中的同名内容。

方案2：Semgrep规则匹配（无需复杂编码，配置简单） #

Semgrep是语法感知的代码搜索替换工具，不需要掌握AST语法，写简单规则就能覆盖绝大多数场景，比纯正则精准很多。

• 安装依赖：pip install semgrep
• 编写规则文件rename_rule.yml：

rules:
  - id: replace-import-statement
    pattern: from a.b import foo_method
    fix: from b.d import bar_method
    languages: [python]
    severity: INFO
  - id: replace-method-call
    pattern: foo_method(...)
    fix: bar_method(...)
    languages: [python]
    severity: INFO

• 执行自动修复：semgrep scan --config rename_rule.yml --autofix 你的源码根目录

方案3：codemod工具（适合简单场景快速使用） #

如果你的代码库导入写法比较统一，没有太复杂的别名、批量导入场景，可以用codemod工具，它比纯正则多了代码上下文感知能力，误改概率更低。

• 安装依赖：npm install -g codemod
• 替换导入语句：codemod -d 你的源码根目录 "from a\.b import (.*)foo_method(.*)" "from b.d import \1bar_method\2" --extension py
• 替换方法调用：codemod -d 你的源码根目录 "foo_method\(" "bar_method(" --extension py

重要提示：无论使用哪种方案，运行前务必先提交所有代码到Git仓库，方便修改出错时快速回滚。

内容的提问来源于stack exchange，提问作者guettli