分享一个提高运维效率的 Python 脚本

17 天前

咸鱼Linux运维

哈喽大家好我是咸鱼，今天给大家分享一个能够提升运维效率的 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]
        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)