很多时候在Linux系统下运行python程序时,控制台会输出一些有用的信息。为了方便保存这些信息,有时需要对这些信息进行保存。这里介绍几种将控制台输出保存到文件中的方式:
1 重定向标准输出流
重定向标准输出流有两种方式,既可以在每个print方法中进行重定向,如下所示:
# assume the log file is 'a.log' # for python2 print a.log, 'print something' # for python3 print('print something', file=a.log)
同时也可以在全局上进行设置:
import sys f = open('a.log', 'a') sys.stdout = f sys.stderr = f # redirect std err, if necessary
2 使用tee命令重定向
上述方法的缺点在于重定向后,控制台就不再显示信息,可能对观察程序现象造成困难。使用tee命令则可以在保存标准输出的同时在控制台上仍然显示信息。使用范例如下:
python a_script.py 2>&1 | tee a.log
这种方法的缺点是控制台显示的内容会断断续续出现,与没有重定向时显示的方式有些不同。
3 自定义logger
我们可以在python程序中自定义一个记录者类,用来同时写文件以及在控制台进行显示。一个例子如下:
import sys class Logger(object): def __init__(self, filename='default.log', stream=sys.stdout): self.terminal = stream self.log = open(filename, 'a') def write(self, message): self.terminal.write(message) self.log.write(message) def flush(self): pass sys.stdout = Logger(a.log, sys.stdout) sys.stderr = Logger(a.log_file, sys.stderr) # redirect std err, if necessary # now it works print 'print something'
此时控制台的显示也同原先一样正常。
以上这篇python将控制台输出保存至文件的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。