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

客官°小女子只卖身不卖艺 2024-03-22 14:27 47阅读 0赞

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

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

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

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

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

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

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

![e84bb31abf37da0dafe563a2ec2a564a.png][]

*  修改一个内容

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

![007077d4397fb6d97b979df37b1c0e42.png][]

*  修改多个内容

python sed.py c:\name.txt Kanye,male  Edison,female

![4893ca113d43f05f3f6dc8767bea52e8.png][]

**源码在文末哦**

首先我们导入模块

import sys
    import os
    import time
    from shutil import copy

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

def data_preprocess():
        file_name, old_str, new_str = sys.argv[1:4] #接收输入的参数：目标文件名，旧内容、新内容、字符编码
        try: # 传入字符编码参数
            encoding = sys.argv[4] 
        except IndexError: # 没有传入字符编码参数的话就默认使用 'utf-8' 字符编码
            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) # 判断用户输入的 old_str 和 new_str 是不是一一对应
        trans_tabs = list(zip(old_str_list, new_str_list)) # 将要旧内容列表(old_str_list)和新内容列表( new_str_list)中的元素一一对应
    
        return file_name, trans_tabs # 返回目标文件名，以及一个旧内容元素和新内容元素一一对应的列表

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

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

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

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

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

file_name = c:\test.txt
    trans_tabs = [(Edsion, Kanye)]

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

python sed.py c:\test.txt edsion,kanye,fish Edsion,Kanye,Fish utf-8

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

file_name = c:\test.txt
    trans_tabs = [(edsion, Edsion), (kanye,Kanye), (fish,Fish)]

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

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)

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

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

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

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.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)

[e84bb31abf37da0dafe563a2ec2a564a.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/22/636eb8fdc0e94baab64eecacf2f621fb.png
[007077d4397fb6d97b979df37b1c0e42.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/22/9c2f14c9723e45ddb8122a12bfbb9367.png
[4893ca113d43f05f3f6dc8767bea52e8.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/22/a25c4bc639c2402daded19d821b4237a.png