Python實現多個檔案中替換字串__Python

import stringimport osimport structimport reimport fileinputimport 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 fileArraydef 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')