python将控制台输出保存至文件的方法_Python

python将控制台输出保存至文件的方法

2021-05-12 00:33KFXW Python

今天小编就为大家分享一篇python将控制台输出保存至文件的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

很多时候在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'