51Testing软件测试论坛

标题: 推荐一个提高运维效率的 Python 脚本 [打印本页]

作者: lsekfe 时间: 2023-6-16 11:00
标题: 推荐一个提高运维效率的 Python 脚本
哈喽大家好我是咸鱼，今天给大家分享一个能够提升运维效率的 python 脚本
　　咸鱼平常在工作当中通常会接触到下面类似的场景：
　　1. 容灾切换的时候批量对机器上的配置文件内容进行修改替换
　　2. 对机器批量替换某个文件中的字段
　　对于 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）中。
　　修改完毕之后，把临时文件重命名一下、把旧目标文件删掉，这个临时文件就变成了修改内容后的目标文件了。
　　源码
　　修改完毕之后，把临时文件重命名一下、把旧目标文件删掉，这个临时文件就变成了修改内容后的目标文件了。
　　mport 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)

欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/)