使用命令行簡單地替換一個文件中的文本內容,并且生成一個新的自定義文件名的文件。這是我們平時工作中常見的一個小任務,下面的這一段小代碼能夠輕松地完成這個任務:
~~~
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:。。。”。