用python3实现linux的sed功能

时间:2021-07-12 19:39:38

sed是一个很好的文件处理工具,本身是一个管道命令,主要是以行为单位进行处理,可以将数据行进行替换、删除、新增、选取等特定工作。现在用python简单实现sed的主要命令,将old_text替换为new_text,同时将原来的文件备份,原有的文件被修改。

sed.py的内容如下:

 import sys
import os
import shutil if len(sys.argv) < 4:
sys.exit("usage:sed.py old_text new_text file_name") # 假定程序的参数是正确的
# 取参数赋值
progran_ame, old_text, new_text, arg_file_name = sys.argv
print(progran_ame, old_text, new_text, arg_file_name) if not os.path.exists(arg_file_name):
sys.exit("文件%s不存在" % arg_file_name) # 判断输入的参数是否为绝对路径, 如果是相对路径则取得绝对路径
if os.path.isabs(arg_file_name):
src_file = arg_file_name
else:
src_file = os.path.realpath(arg_file_name) # 将原来的文件重新命名,得到备份文件名
src_path_name = os.path.dirname(src_file)
src_file_name = os.path.basename(src_file)
bak_file_name = os.path.splitext(src_file_name)[0]+'_bak'+os.path.splitext(src_file_name)[1]
back_file = src_path_name+os.sep+bak_file_name # 备份文件
shutil.copy(src_file, back_file) out_file = open(src_file, 'w', encoding='utf-8')
# 对文件的每一行进行遍历,同时进行替换操作
with open(back_file, encoding='utf-8') as f:
for line in f:
out_file.writelines(line.replace(old_text, new_text)) out_file.close()

测试

test.txt文件内容如下

bj您好
我是小笨蛋
欢迎你……bj

python sed.py bj 北京 test.txt

将test.txt中"bj"替换为"北京"

执行后,生成test_bak.txt文件,内容和test.txt一样。test.txt中的内容变为

北京您好
我是小笨蛋
欢迎你……北京