将所有这些代码放入一个名为的文件中mass_replace
。在Linux或Mac OS X下,您可以执行chmod +x mass_replace
然后运行。在Windows下,您可以运行,python mass_replace
后跟适当的参数。
#!/usr/bin/python
import os
import re
import sys
# list of extensions to replace
DEFAULT_REPLACE_EXTENSIONS = None
# example: uncomment next line to only replace *.c, *.h, and/or *.txt
# DEFAULT_REPLACE_EXTENSIONS = (".c", ".h", ".txt")
def try_to_replace(fname, replace_extensions=DEFAULT_REPLACE_EXTENSIONS):
if replace_extensions:
return fname.lower().endswith(replace_extensions)
return True
def file_replace(fname, pat, s_after):
# first, see if the pattern is even in the file.
with open(fname) as f:
if not any(re.search(pat, line) for line in f):
return # pattern does not occur in file so we are done.
# pattern is in the file, so perform replace operation.
with open(fname) as f:
out_fname = fname + ".tmp"
out = open(out_fname, "w")
for line in f:
out.write(re.sub(pat, s_after, line))
out.close()
os.rename(out_fname, fname)
def mass_replace(dir_name, s_before, s_after, replace_extensions=DEFAULT_REPLACE_EXTENSIONS):
pat = re.compile(s_before)
for dirpath, dirnames, filenames in os.walk(dir_name):
for fname in filenames:
if try_to_replace(fname, replace_extensions):
fullname = os.path.join(dirpath, fname)
file_replace(fullname, pat, s_after)
if len(sys.argv) != 4:
u = "Usage: mass_replace <dir_name> <string_before> <string_after>\n"
sys.stderr.write(u)
sys.exit(1)
mass_replace(sys.argv[1], sys.argv[2], sys.argv[3])
编辑:我已经从原始答案更改了上面的代码。有几个变化。首先,mass_replace()
现在调用re.compile()
预编译搜索模式;第二,要检查文件的扩展名,我们现在将文件扩展名的元组传递给,.endswith()
而不是调用.endswith()
三次。第三,它现在使用with
最新版本的Python中可用的语句;最后,file_replace()
现在检查是否在文件中找到了模式,如果未找到模式,则不重写文件。(旧版本将重写每个文件,即使输出文件与输入文件相同,也会更改时间戳;这太小了。)
编辑:我将其更改为默认值,以替换每个文件,但是您可以编辑一行以将其限制为特定的扩展名。我认为替换每个文件是一个更有用的现成默认值。可以使用扩展名列表或不可触摸的文件名,使其不区分大小写的选项等扩展。
编辑:在评论中,@ asciimo指出了一个错误。我对此进行了编辑以修复该错误。str.endswith()
被证明可以接受尝试的字符串元组,但不能接受列表。固定。另外,我使几个函数接受一个可选参数,以使您可以传入一个扩展元组。修改它以接受命令行参数来指定扩展名应该很容易。