Python实现多个文件中替换字符串

import string
import os
import struct
import re
import fileinput
import fnmatch

这里主要使用了fileinput和fnmatch两个Module

fnmatch.fnmatch主要用来判断文件的后缀名

fnmatch. fnmatch ( filename, pattern )对于pattern:

*.txt →TXT文件

*.doc  →DOC文件

*.* →所有文件

fileinput中的inplace=True表示把输出重定向到文件。这样所有print输出都是指向文件的。

def walkDir(directory, ext='*.*', topdown=True):
    fileArray = []
    for root, dirs, files in os.walk(directory, topdown):
        for name in files:
            if fnmatch.fnmatch(name, ext):
                fileArray.append(os.path.abspath(os.path.join(root, name)))
    return fileArray

def replaceInFile(filename, strFrom, strTo):
    for line in fileinput.input(filename, inplace=True):
        if re.search(strFrom, line):
            line = line.replace(strFrom, strTo)
        print line,

if __name__ == '__main__':
    for filename in walkDir('.', '*.txt'):
        replaceInFile(filename, 'abc', 'cba')