替换文件内指定字符串

最近需要对文件内的某字符串（source）替换成另一字符串（target）。

手动改其实也很简单方便，但是简单耐不住文件多。当需要二三十个sql文件需要替换里面的字符串（source）为字符串（target）时，写个脚本是个不错的想法。尤其是，每个文件需要替换的字符串是相同的。

如下：

单字串符替换

def replace_txt(file, source, target):
"""替换文件后缀为.sql 文件内的source字符为target字符。文件名在原有基础上加 _replace"""
# 读取文件
with open(file, "r") as f:
# 写入文件
with open(os.path.basename(file)[:-4] + "_replace.sql", "w", encoding="utf-8") as fw:
for each_line in f.readlines():
# 若读取行为空，则不作任何操作
if not each_line:
continue
# 替换字符串
fw.write(each_line.replace(source, target))

source为需要替换的字符串，target是替换后的字符串。按照上述定义的函数replace_txt（），读取每一行文件，然后逐行进行替换操作。并写入新文件。

例如：原文件为a.sql，source为：abcd, target为：aaaa

执行上述函数后，结果会生成一个 a_replace.sql文件，文件内的所有“abcd”会替换成“aaaa”。

如果文件名后缀不是三个字符组成的，新文件命名需要稍改一下。

如果文件名没有“.”字符，可以将

os.path.basename(file)[:-4] + "_replace.sql"
改成
os.path.basename(file).split(".")[0] + "_replace." + os.path.basename(file).split(".")[1]

上述程序会生成新文件，如果有需要，可以删除原文件，将新生成的文件命名为原文件名。

多字符串替换

如果出现一个文件内需要替换多个字符串，上述脚本显然不符合要求。要写出一个高可用的脚本，还需要进一步处理。如下：

def replace_txt(file, **kwargs):
"""替换文件后缀为.sql 文件内的source字符为target字符。文件名在原有基础上加 _replace"""
# 读取文件
with open(file, "r") as f:
# 写入文件
with open(os.path.basename(file).split(".")[0] + "_replace." + os.path.basename(file).split(".")[1], "w", encoding="utf-8") as fw:
for each_line in f.readlines():
# 若读取行为空，则不作任何操作
if not each_line:
continue
# 替换字符串
for source,target in kwargs.items():
fw.write(each_line.replace(source, target))

原来的source，target入参改成 **kwargs以字典项形式传入。用

for source,target in kwargs.items():
fw.write(each_line.replace(source, target))

方法读取所有需要替换的字符串，key为需替换字符串，value为替换后字符串。

实际操作一把：

有demo.txt文件，内容如下：

abcd abcd abcd abcd
abcde abcde abcde abcde
abcdef abcdef abcdef abcdef

执行脚本命令如下：

replace_txt(r"C:\Users\Administrator\Desktop\scripts\demo.txt", abc="ABC", f="F")

得到结果文件demo_replace.txt内容为：

ABCd ABCd ABCd ABCd
abcd abcd abcd abcd
ABCde ABCde ABCde ABCde
abcde abcde abcde abcde
ABCdef ABCdef ABCdef ABCdefabcdeF abcdeF abcdeF abcdeF

一切ok。