用python3实现linux的sed功能

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中的内容变为

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