python小技巧之批量修改文件中的字符串

import os

def replace_in_file(file_path, old_string, new_string):
    with open(file_path, 'r', encoding='utf-8') as file:
        file_data = file.read()
    
    new_data = file_data.replace(old_string, new_string)
    
    with open(file_path, 'w', encoding='utf-8') as file:
        file.write(new_data)

def batch_replace_in_directory(directory, old_string, new_string, extension=".v"):
    for root, _, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                file_path = os.path.join(root, file)
                replace_in_file(file_path, old_string, new_string)
                print(f"Replaced in {file_path}")

# 获取当前脚本所在的绝对路径
current_directory = os.path.abspath(os.path.dirname(__file__))
old_string = "OLD"
new_string = "NEW"

# 批量替换
batch_replace_in_directory(current_directory, old_string, new_string)