分享一个快速提高运维效率的 Python 脚本，赶紧码住！

本文链接：https://blog.csdn.net/Pythonxiaoxin6/article/details/135690018

本文介绍了一个Python脚本，用于在Linux和Windows机器上批量替换文本文件中的指定内容，支持多内容替换，并提供了代码示例和备份功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

哈喽大家好，今天给大家分享一个能够提升运维效率的 python 脚本

我们平常在工作当中通常会接触到下面类似的场景：

容灾切换的时候批量对机器上的配置文件内容进行修改替换
对机器批量替换某个文件中的字段

对于 Linux 机器，咸鱼可以写个 shell 脚本或者直接批量使用 sed 命令就能很好地解决

但对于 Windows 机器，上面的方法就不管用了，我们就需要想其他的办法

这里咸鱼给大家分享一个由 python 编写的脚本，这个脚本能够去替换指定文本文件中的指定内容，而且还能一次替换多个内容

我们先看效果，目标文件：name.txt

修改一个内容

修改多个内容

源码在文末哦！****

首先我们导入模块

然后我们来看一下脚本中的第一个函数——data_preprocess()

这个函数实现的功能是接收输入的参数（目标文件名、要替换的内容，替换的内容、字符编码格式），然后将要替换的内容与替换的内容分别转换成列表形式

注意：替换多个内容时在多个内容之间用逗号隔开

例如我们敲如下命令替换一个内容：

python sed.py c:\test.txt Edison Kanye

这个函数就会返回下面内容（由于命令里没有传入字符编码参数，采取默认的 UTF-8）

如果我们需要替换多个内容（例如将首字母改成大写）

这个函数就会返回下面内容（命令传入了字符编码参数）

接下来我们来看第二个函数——backup()

这个函数的功能就是在修改文件之前先把文件备份，防止后期我们需要回滚复原

最后我们来看最后一个函数—— sed()

这个函数便是整个脚本的核心，它负责去执行修改替换文件内容的操作

首先先打开一个临时文件（file_name.swap）和目标文件（file_name）

然后对目标文件（file_name）进行逐行读取到内存上，再去对内容进行修改，最后将修改后的内容写入到这个临时文件（file_name.swap）中

修改完毕之后，把临时文件重命名一下、把旧目标文件删掉，这个临时文件就变成了修改内容后的目标文件了

源码修改完毕之后，把临时文件重命名一下、把旧目标文件删掉，这个临时文件就变成了修改内容后的目标文件了

import sys
import os
import time
from shutil import copy


def data_preprocess():
    file_name, old_str, new_str = sys.argv[1:4]
    try:
        encoding = sys.argv[4]
    except IndexError:
        encoding = 'utf-8'

    old_str_list = [i.encode(encoding) for i in old_str.split(',')]
    new_str_list = [i.encode(encoding) for i in new_str.split(',')]

    assert len(old_str_list) == len(new_str_list)
    trans_tabs = list(zip(old_str_list, new_str_list))

    return file_name, trans_tabs


def backup(file_name):
    time_mark = time.strftime('%Y%m%d_%H%M%S')
    bak_dir = r'C:\Users\Administrator\Desktop'
    basename = os.path.basename(file_name)
    os.makedirs(bak_dir) if not os.path.isdir(bak_dir) else True
    copy(file_name, os.path.join(bak_dir, basename + '_' + time_mark))
    print("备份 %s" %file_name)


def sed(file_name, trans_tabs):
    with open(file_name + '.swap', 'wb') as swap_fs, open(file_name, 'rb') as file_names:
        for line in file_names:
            for tab in trans_tabs:
                line = line.replace(tab[0], tab[1]) if tab[0] in line else line
            swap_fs.write(line)
    os.remove(file_name)
    os.rename(file_name + '.swap', file_name)


file_name, trans_tabs = data_preprocess()
backup(file_name)
sed(file_name, trans_tabs)

以上就是今天的全部内容分享，觉得有用的话欢迎点赞收藏哦！

Python经验分享

学好 Python 不论是用于就业还是做副业赚钱都不错，而且学好Python还能契合未来发展趋势——人工智能、机器学习、深度学习等。
小编是一名Python开发工程师，自己整理了一套最新的Python系统学习教程，包括从基础的python脚本到web开发、爬虫、数据分析、数据可视化、机器学习等。如果你也喜欢编程，想通过学习Python转行、做副业或者提升工作效率，这份【最新全套Python学习资料】一定对你有用！

包括：Python激活码+安装包、Python web开发，Python爬虫，Python数据分析，人工智能、机器学习、Python量化交易等学习教程。带你从零基础系统性的学好Python！