咸鱼平常在工作当中通常会接触到下面类似的场景：
- 容灾切换的时候批量对机器上的配置文件内容进行修改替换
- 对机器批量替换某个文件中的字段
但对于 Windows 机器，上面的方法就不管用了，我们就需要想其他的办法
替换指定文本文件中的指定内容，而且还能一次替换多个内容
- 修改一个内容
python sed.py c:\name.txt Kanye Edison
- 修改多个内容
python sed.py c:\name.txt Kanye,male Edison,female
源码在文末哦
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）
修改完毕之后，把临时文件重命名一下、把旧目标文件删掉，这个临时文件就变成了修改内容后的目标文件了
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