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'

此時控制檯的顯示也同原先一樣正常。

以上這篇python將控制檯輸出保存至文件的方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持神馬文庫。