最近需要对文件内的某字符串(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。