Python上下文管理器

本節(jié)嚴(yán)格意義上并非新的重定向方式，而是利用Pyhton上下文管理器優(yōu)化上節(jié)的代碼實(shí)現(xiàn)。借助于上下文管理器語(yǔ)法，可不必向重定向使用者暴露sys.stdout。

首先考慮輸出抑制，基于上下文管理器語(yǔ)法實(shí)現(xiàn)如下：

importsys,cStringIO,contextlib

classDummyFile:

defwrite(self,outStr):pass

@contextlib.contextmanager

defMuteStdout():

savedStdout=sys.stdout

sys.stdout=cStringIO.StringIO()#DummyFile()

try:

yield

exceptException:#捕獲到錯(cuò)誤時(shí)，屏顯被抑制的輸出(該處理并非必需)

content,sys.stdout=sys.stdout,savedStdout

printcontent.getvalue()#;raise

#finally:

sys.stdout=savedStdout

使用示例如下：

withMuteStdout():

print"I'llshowupwhenisexecuted!"#不屏顯不寫(xiě)入

raise#屏顯上句

print"I'mhidingmyselfsomewhere:)"#不屏顯

再考慮更通用的輸出重定向：

importos,sys

fromcontextlibimportcontextmanager

@contextmanager

defRedirectStdout(newStdout):

savedStdout,sys.stdout=sys.stdout,newStdout

try:

yield

finally:

sys.stdout=savedStdout

使用示例如下：

defGreeting():print'Hello,boss!'

withopen('out.txt',"w+")asfile:

print"I'mwritingtoyou..."#屏顯

withRedirectStdout(file):

print'Ihopethisletterfindsyouwell!'#寫(xiě)入文件

print'Checkyourmailbox.'#屏顯

withopen(os.devnull,"w+")asfile,RedirectStdout(file):

Greeting()#不屏顯不寫(xiě)入

print'Ideserveapayraise:)'#不屏顯不寫(xiě)入

print'DidyouhearwhatIsaid?'#屏顯

可見(jiàn)，with內(nèi)嵌塊里的函數(shù)和print語(yǔ)句輸出均被重定向。注意，上述示例不是線程安全的，主要適用于單線程。

當(dāng)函數(shù)被頻繁調(diào)用時(shí)，建議使用裝飾器包裝該函數(shù)。這樣，僅需修改該函數(shù)定義，而無(wú)需在每次調(diào)用該函數(shù)時(shí)使用with語(yǔ)句包裹。示例如下：

importsys,cStringIO,functools

defMuteStdout(retCache=False):

defdecorator(func):

@functools.wraps(func)

defwrapper(*args,**kwargs):

savedStdout=sys.stdout

sys.stdout=cStringIO.StringIO()

try:

ret=func(*args,**kwargs)

ifretCache==True:

ret=sys.stdout.getvalue().strip()

finally:

sys.stdout=savedStdout

returnret

returnwrapper

returndecorator

若裝飾器MuteStdout的參數(shù)retCache為真，外部調(diào)用func()函數(shù)時(shí)將返回該函數(shù)內(nèi)部print輸出的內(nèi)容(可供屏顯);若retCache為假，外部調(diào)用func()函數(shù)時(shí)將返回該函數(shù)的返回值(抑制輸出)。

MuteStdout裝飾器使用示例如下：

@MuteStdout(True)

defExclaim():print'Iamproudofmyself!'

@MuteStdout()

defMumble():print'Ilackconfidence...';return'sad'

printExclaim(),Exclaim.__name__#屏顯'Iamproudofmyself!Exclaim'

printMumble(),Mumble.__name__#屏顯'sadMumble'

在所有線程中，被裝飾函數(shù)執(zhí)行期間，sys.stdout都會(huì)被MuteStdout裝飾器劫持。而且，函數(shù)一經(jīng)裝飾便無(wú)法移除裝飾。因此，使用該裝飾器時(shí)應(yīng)慎重考慮場(chǎng)景。

接著，考慮創(chuàng)建RedirectStdout裝飾器：

defRedirectStdout(newStdout=sys.stdout):

defdecorator(func):

defwrapper(*args,**kwargs):

savedStdout,sys.stdout=sys.stdout,newStdout

try:

returnfunc(*args,**kwargs)

finally:

sys.stdout=savedStdout

returnwrapper

returndecorator

使用示例如下：

file=open('out.txt',"w+")

@RedirectStdout(file)

defFunNoArg():print'Noargument.'

@RedirectStdout(file)

defFunOneArg(a):print'Oneargument:',a

defFunTwoArg(a,b):print'Twoarguments:%s,%s'%(a,b)

FunNoArg()#寫(xiě)文件'Noargument.'

FunOneArg(1984)#寫(xiě)文件'Oneargument:1984'

RedirectStdout()(FunTwoArg)(10,29)#屏顯'Twoarguments:10,29'

printFunNoArg.__name__#屏顯'wrapper'(應(yīng)顯示'FunNoArg')

file.close()

注意FunTwoArg()函數(shù)的定義和調(diào)用與其他函數(shù)的不同，這是兩種等效的語(yǔ)法。此外，RedirectStdout裝飾器的最內(nèi)層函數(shù)wrapper()未使用"functools.wraps(func)"修飾，會(huì)丟失被裝飾函數(shù)原有的特殊屬性(如函數(shù)名、文檔字符串等)。

以上內(nèi)容為大家介紹了Python上下文管理器，希望對(duì)大家有所幫助，如果想要了解更多Python相關(guān)知識(shí)，請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。