Python3搜索及替换文件中文本的方法

# 将文件中的某个字符串改变成另一个 
# 下面代码实现从一个特定文件或标准输入读取文件，
# 然后替换字符串，然后写入一个指定的文件 
import os, sys
nargs = len(sys.argv)
if not 3 <= nargs <= 5:
  print('usage: %s search_text repalce_text [infile [outfile]]' % \
     os.path.basename(sys.argv[0]))
else:
  search_text = sys.argv[1]
  replace_text = sys.argv[2]
  input_file = sys.stdin
  output_file = sys.stdout
  if nargs > 3:
    input_file = open(sys.argv[3])
  if nargs > 4:
    output_file = open(sys.argv[4], 'w')
  for s in input_file:
    output_file.write(s.replace(search_text, replace_text))
  output_file.close()
  input_file.close()