Python菜譜13：在文件中搜索以及替換文本

使用命令行簡單地替換一個文件中的文本內(nèi)容，并且生成一個新的自定義文件名的文件。這是我們平時工作中常見的一個小任務(wù)，下面的這一段小代碼能夠輕松地完成這個任務(wù):

import os
import sys
usage = "usage: %s search_text replace_text [infilename [outfilename]]" % os.path.basename(
    sys.argv[0])

if len(sys.argv) < 3:
    print usage

else:
    stext = sys.argv[1]
    rtext = sys.argv[2]
    print "There are %s args " % len(sys.argv)

    if len(sys.argv) > 4:
        input = open(sys.argv[3])
        output = open(sys.argv[4], 'w')

        for s in input:
            output.write(s.replace(stext, rtext))

        input.close()
        output.close()

當(dāng)我們使用 “python cookbook_13.py 1 a test.txt new.txt” 命令行的時候，test.txt 中 1 會被替換成 a，并且替換后的內(nèi)容寫入到 new.txt 中。

注意：infilename，outfilename 這兩個參數(shù)沒有的話，程序并不會報錯，但是會輸出類似 “There are ...” 的語句。如果命令行參數(shù)小于 3 的話，會輸出 “usage:。。?！薄?/p>