再考慮更通用的輸出重定向:
import os, sys
from contextlib import contextmanager
@contextmanager
def RedirectStdout(newStdout):
savedStdout, sys.stdout = sys.stdout, newStdout
try:
yield
finally:
sys.stdout = savedStdout
使用示例如下:
def Greeting(): print 'Hello, boss!'
with open('out.txt', "w+") as file:
print "I'm writing to you..." #屏顯
with RedirectStdout(file):
print 'I hope this letter finds you well!' #寫入文件
print 'Check your mailbox.' #屏顯
with open(os.devnull, "w+") as file, RedirectStdout(file):
Greeting() #不屏顯不寫入
print 'I deserve a pay raise:)' #不屏顯不寫入
print 'Did you hear what I said?' #屏顯
可見,with內(nèi)嵌塊里的函數(shù)和print語(yǔ)句輸出均被重定向。注意,上述示例不是線程安全的,主要適用于單線程。
當(dāng)函數(shù)被頻繁調(diào)用時(shí),建議使用裝飾器包裝該函數(shù)。這樣,僅需修改該函數(shù)定義,而無(wú)需在每次調(diào)用該函數(shù)時(shí)使用with語(yǔ)句包裹。示例如下:
import sys, cStringIO, functools
def MuteStdout(retCache=False):
def decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
savedStdout = sys.stdout
sys.stdout = cStringIO.StringIO()
try:
ret = func(*args, **kwargs)
if retCache == True:
ret = sys.stdout.getvalue().strip()
finally:
sys.stdout = savedStdout
return ret
return wrapper
return decorator
若裝飾器MuteStdout的參數(shù)retCache為真,外部調(diào)用func()函數(shù)時(shí)將返回該函數(shù)內(nèi)部print輸出的內(nèi)容(可供屏顯);若retCache為假,外部調(diào)用func()函數(shù)時(shí)將返回該函數(shù)的返回值(抑制輸出)。
MuteStdout裝飾器使用示例如下:
@MuteStdout(True)
def Exclaim(): print 'I am proud of myself!'
@MuteStdout()
def Mumble(): print 'I lack confidence...'; return 'sad'
print Exclaim(), Exclaim.__name__ #屏顯'I am proud of myself! Exclaim'
print Mumble(), Mumble.__name__ #屏顯'sad Mumble'
在所有線程中,被裝飾函數(shù)執(zhí)行期間,sys.stdout都會(huì)被MuteStdout裝飾器劫持。而且,函數(shù)一經(jīng)裝飾便無(wú)法移除裝飾。因此,使用該裝飾器時(shí)應(yīng)慎重考慮場(chǎng)景。
接著,考慮創(chuàng)建RedirectStdout裝飾器:
def RedirectStdout(newStdout=sys.stdout):
def decorator(func):
def wrapper(*args,**kwargs):
savedStdout, sys.stdout = sys.stdout, newStdout
try:
return func(*args, **kwargs)
finally:
sys.stdout = savedStdout
return wrapper
return decorator
使用示例如下:
file = open('out.txt', "w+")
@RedirectStdout(file)
def FunNoArg(): print 'No argument.'
@RedirectStdout(file)
def FunOneArg(a): print 'One argument:', a
def FunTwoArg(a, b): print 'Two arguments: %s, %s' %(a,b)
FunNoArg() #寫文件'No argument.'
FunOneArg(1984) #寫文件'One argument: 1984'
RedirectStdout()(FunTwoArg)(10,29) #屏顯'Two arguments: 10, 29'
print FunNoArg.__name__ #屏顯'wrapper'(應(yīng)顯示'FunNoArg')
file.close()
注意FunTwoArg()函數(shù)的定義和調(diào)用與其他函數(shù)的不同,這是兩種等效的語(yǔ)法。此外,RedirectStdout裝飾器的最內(nèi)層函數(shù)wrapper()未使用"functools.wraps(func)"修飾,會(huì)丟失被裝飾函數(shù)原有的特殊屬性(如函數(shù)名、文檔字符串等)。
2.5 logging模塊重定向
對(duì)于代碼量較大的工程,建議使用logging模塊進(jìn)行輸出。該模塊是線程安全的,可將日志信息輸出到控制臺(tái)、寫入文件、使用TCP/UDP協(xié)議發(fā)送到網(wǎng)絡(luò)等等。
默認(rèn)情況下logging模塊將日志輸出到控制臺(tái)(標(biāo)準(zhǔn)出錯(cuò)),且只顯示大于或等于設(shè)置的日志級(jí)別的日志。日志級(jí)別由高到低為CRITICAL > ERROR > WARNING > INFO > DEBUG > NOTSET,默認(rèn)級(jí)別為WARNING。
以下示例將日志信息分別輸出到控制臺(tái)和寫入文件:
import logging
logging.basicConfig(level = logging.DEBUG,
format = '%(asctime)s [%(levelname)s] at %(filename)s,%(lineno)d: %(message)s',
datefmt = '%Y-%m-%d(%a)%H:%M:%S',
filename = 'out.txt',
filemode = 'w')
#將大于或等于INFO級(jí)別的日志信息輸出到StreamHandler(默認(rèn)為標(biāo)準(zhǔn)錯(cuò)誤)
console = logging.StreamHandler()
console.setLevel(logging.INFO)
formatter = logging.Formatter('[%(levelname)-8s] %(message)s') #屏顯實(shí)時(shí)查看,無(wú)需時(shí)間
console.setFormatter(formatter)
logging.getLogger().addHandler(console)
logging.debug('gubed'); logging.info('ofni'); logging.critical('lacitirc')
通過(guò)對(duì)多個(gè)handler設(shè)置不同的level參數(shù),可將不同的日志內(nèi)容輸入到不同的地方。本例使用在logging模塊內(nèi)置的StreamHandler(和FileHandler),運(yùn)行后屏幕上顯示:
[INFO ] ofni
[CRITICAL] lacitirc
out.txt文件內(nèi)容則為:
2016-05-13(Fri)17:10:53 [DEBUG] at test.py,25: gubed
2016-05-13(Fri)17:10:53 [INFO] at test.py,25: ofni
2016-05-13(Fri)17:10:53 [CRITICAL] at test.py,25: lacitirc
評(píng)論
查看更多