本文已参与「新人创作礼」活动，一起开启掘金创作之路。

 前言

接上文 批量修改文件内容

一、为什么选择 python

1. 近年来 Python 的使用率越来越高，甚至热度已经超过了 java 
2. 相比 shell 脚本，python 语法简单，更容易阅读，内置了许多第三方模块可供使用
3. 相比 java 等其他语言，python 易于发布部署，更适用于一些日常脚本的编写

二、使用步骤

1.搜索

采用递归的方式，遍历目录下的所有文件

# 文件查找 find . -name file_name -type f
# 查找函数：search_path 查找根路径 file_name 需要查找的文件名
def search(search_path, search_file_name, search_result):
    # 获取当前路径下地所有文件
    all_file = os.listdir(search_path)
    # 对于每一个文件
    for each_file in all_file:
        # 若文件为一个文件夹
        if os.path.isdir(search_path + os.sep + each_file):
            # 递归查找
            search(search_path + os.sep + each_file, search_file_name, search_result)
        # 如果是需要被查找的文件
        elif each_file == search_file_name:
            # 输出路径
            search_result.append(search_path + os.sep + search_file_name)

2.替换

代码如下：

# 替换 sed -i 's/old_str/new_str/'
# 文本替换 replace_file_name 需要替换的文件路径，replace_old_str 要替换的字符，replace_new_str 替换的字符
def replace(replace_file_name, replace_old_str, replace_new_str):
    f1 = open(replace_file_name, "r")
    content = f1.read()
    f1.close()
    t = content.replace(replace_old_str, replace_new_str)
    with open(replace_file_name, "w") as f2:
        f2.write(t)
    f2.close()

3 整合

用到了 getopt 模块

if __name__ == '__main__':
    result = []
    # 默认当前目录
    path = os.getcwd()
    file_name = "test"
    old_str = "old_str"
    new_str = "new_str"
    try:
        opts, args = getopt.getopt(sys.argv[1:], "hp:f:o:n:", ["help", "path=", "file=", "old=", "new="])
    except getopt.GetoptError:
        print('usage: search_and_replace.py -p <path> -f <file_name> -o <old_str> -n <new_str>')
        sys.exit(2)
    for opt, arg in opts:
        if opt in ('-h', '--help'):
            print('usage: search_and_replace.py -p <path> -f <file_name> -o <old_str> -n <new_str>')
            sys.exit()
        elif opt in ("-p", "--path"):
            path = arg
        elif opt in ("-f", "--file"):
            file_name = arg
        elif opt in ("-o", "--old"):
            old_str = arg
        elif opt in ("-n", "--new"):
            new_str = arg
    search(path, file_name, result)
    for file_name in result:
        replace(file_name, old_str, new_str)
        print("replace {} to {} in file {} successfully".format(old_str, new_str, file_name))

4 脚本使用

将文件名为 config 的文件中的 old_str 转换为 new_str

python search_and_replace.py -f config -o old_str -n new_str 

结果：