Question

关于python的输出重定向

如何用python实现输出重定向到标准输出和文件两个地方，就像linux 的tee命令一样

Answer 1

import sys
f = open('a.txt','w')
print >>sys.stdout,'hello,world'
hello,world
print >>f,'hello,world'
f.close()

输出到屏幕的内容重定向到文件，供参考。

另，print函数的源码

def print(stream):
  """ print(value, ..., sep=' ', end='\\n', file=sys.stdout)
  
  Prints the values to a stream, or to sys.stdout by default.
  Optional keyword arguments:
  file: a file-like object (stream); defaults to the current sys.stdout.
  sep:  string inserted between values, default a space.
  end:  string appended after the last value, default a newline. """
  pass

Answer 2

判断是否有argv

>>> import sys
>>> print sys.argv
['']

比如假如你的脚本:  cmd [-i file] 这样子的. 那么判断是否传递了 file 这个参数,  有么就打开文件, 没有就从stdin/stdout里读写.

追问

是要同时输出到屏幕上和log文件里的

追答

... 汗... 那还不一样?

你不会只会写入一个文件吧?

import sys
s = 'bala...bala...bala...'

fp = open('./log', 'w')
fp.write(s)
fp.close()
sys.stdout.write(s)

...