使用python怎么将控制台输出保存至文件

1 重定向标准输出流

重定向标准输出流有两种方式,既可以在每个print方法中进行重定向,如下所示:


 
  1. # assume the log file is 'a.log'
  2. # for python2
  3. print >> a.log, 'print something'
  4. # for python3
  5. print('print something', file=a.log)

同时也可以在全局上进行设置:


 
  1. import sys
  2. f = open('a.log', 'a')
  3. sys.stdout = f
  4. sys.stderr = f # redirect std err, if necessary

2 使用tee命令重定向

上述方法的缺点在于重定向后,控制台就不再显示信息,可能对观察程序现象造成困难。使用tee命令则可以在保存标准输出的同时在控制台上仍然显示信息。使用范例如下:


 
  1. python a_script.py 2>&1 | tee a.log

这种方法的缺点是控制台显示的内容会断断续续出现,与没有重定向时显示的方式有些不同。

3 自定义logger

我们可以在python程序中自定义一个记录者类,用来同时写文件以及在控制台进行显示。一个例子如下:


 
  1. import sys
  2. class Logger(object):
  3.  def __init__(self, filename='default.log', stream=sys.stdout):
  4.   self.terminal = stream
  5.   self.log = open(filename, 'a')
  6.  
  7.  def write(self, message):
  8.   self.terminal.write(message)
  9.   self.log.write(message)
  10.  
  11.  def flush(self):
  12.   pass
  13.  
  14. sys.stdout = Logger(a.log, sys.stdout)
  15. sys.stderr = Logger(a.log_file, sys.stderr) # redirect std err, if necessary
  16.  
  17. # now it works
  18. print 'print something'

你可能感兴趣的