如何在PyCharm中批量为多个脚本添加代码修改？

嗨，这个场景太戳人了——85个脚本挨个改简直是体力活！不用慌，给你几个实用的方法，都不用逐个打开文件：

方法一：PyCharm全局正则替换（最便捷） #

这是PyCharm自带的功能，搞定大部分常规需求完全够用：

• 按下Ctrl+Shift+R（Windows/Linux）或Cmd+Shift+R（Mac）调出全局替换窗口
• 在「查找」框里设置定位规则，比如：
  • 要在所有脚本开头加代码：查找框填^（正则表示行首），替换框写你要添加的代码\n$0（$0会保留原来的文件内容）
  • 要在某行代码之后加（比如所有from robot.api import *后面）：查找框填from robot.api import *，替换框写from robot.api import *\n你要添加的代码
• 勾选「正则表达式」选项，然后在「范围」里选择存放85个脚本的文件夹
• 先点击「预览」确认所有要修改的文件都正确匹配，没问题再点「替换全部」

方法二：自制Python脚本批量处理（适合复杂需求） #

如果你的添加逻辑比较特殊（比如要根据文件内容判断插入位置），自己写个小脚本更灵活：

import os

# 替换成你的脚本文件夹路径
target_folder = "/Users/xxx/automation_scripts"
# 替换成你要添加的代码块
code_snippet = """
# 新增的代码块
from custom_utils import log_setup
log_setup.init_logger()
"""

# 遍历文件夹下所有py文件
for file_name in os.listdir(target_folder):
    if file_name.endswith(".py"):
        file_path = os.path.join(target_folder, file_name)
        # 读取原文件内容
        with open(file_path, "r", encoding="utf-8") as f:
            content = f.read()
        # 示例：在文件开头插入代码（你可以改成其他插入逻辑，比如找到特定行后插入）
        new_content = code_snippet.strip() + "\n\n" + content
        # 写入修改后的内容
        with open(file_path, "w", encoding="utf-8") as f:
            f.write(new_content)

👉 注意：先拿1-2个脚本测试，确认逻辑正确后再批量执行，最好先备份所有文件哦！

方法三：PyCharm结构搜索替换（精准匹配代码结构） #

如果要基于代码结构插入（比如在所有测试函数开头加代码），这个高级功能能避免误改：

• 打开Edit > Find > Structural Replace
• 点击「Create Template」，定义搜索模板，比如匹配所有测试函数：

  def $test_func$():
      $body$
  

• 再定义替换模板，把要加的代码插入到函数体内：

  def $test_func$():
      # 新增的前置代码
      setup_test_env()
      $body$
  

• 设置搜索范围为目标文件夹，点击「Find」预览，确认后执行替换

⚠️ 重要提醒：不管用哪种方法，先备份全部脚本或者用Git做版本控制，万一改出问题能快速回滚！

备注：内容来源于stack exchange，提问作者ADS KUL