加载中...

菜谱 13:在文件中搜索以及替换文本


使用命令行简单地替换一个文件中的文本内容,并且生成一个新的自定义文件名的文件。这是我们平时工作中常见的一个小任务,下面的这一段小代码能够轻松地完成这个任务:

  1. import os
  2. import sys
  3. usage = "usage: %s search_text replace_text [infilename [outfilename]]" % os.path.basename(
  4. sys.argv[0])
  5. if len(sys.argv) < 3:
  6. print usage
  7. else:
  8. stext = sys.argv[1]
  9. rtext = sys.argv[2]
  10. print "There are %s args " % len(sys.argv)
  11. if len(sys.argv) > 4:
  12. input = open(sys.argv[3])
  13. output = open(sys.argv[4], 'w')
  14. for s in input:
  15. output.write(s.replace(stext, rtext))
  16. input.close()
  17. output.close()

当我们使用 “python cookbook_13.py 1 a test.txt new.txt” 命令行的时候,test.txt 中 1 会被替换成 a,并且替换后的内容写入到 new.txt 中。

注意:infilename,outfilename 这两个参数没有的话,程序并不会报错,但是会输出类似 “There are ...” 的语句。如果命令行参数小于 3 的话,会输出 “usage:。。。”。


还没有评论.