在檔案中搜尋以及替換文字

使用命令列簡單地替換一個檔案中的文字內容，並且生成一個新的自定義檔名的檔案。這是我們平時工作中常見的一個小任務，下面的這一段小程式碼能夠輕鬆地完成這個任務:

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()

當我們使用 「python cookbook_13.py 1 a test.txt new.txt」 命令列的時候，test.txt 中 1 會被替換成 a，並且替換後的內容寫入到 new.txt 中。

注意：infilename，outfilename 這兩個引數沒有的話，程式並不會報錯，但是會輸出類似 「There are ...」 的語句。如果命令列引數小於 3 的話，會輸出 「usage:。。。」。