Python重定向之print >>和sys.stdout

print>>重定向

這種方式基于print語句的擴(kuò)展形式，即"printobj>>expr"。其中，obj為一個(gè)file-like(尤其是提供write方法的)對(duì)象，為None時(shí)對(duì)應(yīng)標(biāo)準(zhǔn)輸出(sys.stdout)。expr將被輸出到該文件對(duì)象中。示例如下：

memo=cStringIO.StringIO();serr=sys.stderr;file=open('out.txt','w+')

print>>memo,'StringIO';print>>serr,'stderr';print>>file,'file'

print>>None,memo.getvalue()

上述代碼執(zhí)行后，屏顯為"serr"和"StringIO"(兩行，注意順序)，out.txt文件內(nèi)寫入"file"。

可見，這種方式非常靈活和方便。缺點(diǎn)是不適用于輸出語句較多的場(chǎng)景。

sys.stdout重定向

將一個(gè)可寫對(duì)象(如file-like對(duì)象)賦給sys.stdout，可使隨后的print語句輸出至該對(duì)象。重定向結(jié)束后，應(yīng)將sys.stdout恢復(fù)最初的缺省值，即標(biāo)準(zhǔn)輸出。

簡單示例如下：

importsys

savedStdout=sys.stdout#保存標(biāo)準(zhǔn)輸出流

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

sys.stdout=file#標(biāo)準(zhǔn)輸出重定向至文件

print'Thismessageisforfile!'

sys.stdout=savedStdout#恢復(fù)標(biāo)準(zhǔn)輸出流

print'Thismessageisforscreen!'

注意，IDLE中sys.stdout初值為PseudoOutputFile對(duì)象，與sys.__stdout__并不相同。為求通用，本例另行定義變量(savedStdout)保存sys.stdout，下文也將作此處理。此外，本例不適用于經(jīng)由fromsysimportstdout導(dǎo)入的stdout對(duì)象。

以下將自定義多種具有write()方法的file-like對(duì)象，以滿足不同需求：

classRedirectStdout:#importos,sys,cStringIO

def__init__(self):

self.content=''

self.savedStdout=sys.stdout

self.memObj,self.fileObj,self.nulObj=None,None,None

#外部的print語句將執(zhí)行本write()方法，并由當(dāng)前sys.stdout輸出

defwrite(self,outStr):

#self.content.append(outStr)

self.content+=outStr

deftoCons(self):#標(biāo)準(zhǔn)輸出重定向至控制臺(tái)

sys.stdout=self.savedStdout#sys.__stdout__

deftoMemo(self):#標(biāo)準(zhǔn)輸出重定向至內(nèi)存

self.memObj=cStringIO.StringIO()

sys.stdout=self.memObj

deftoFile(self,file='out.txt'):#標(biāo)準(zhǔn)輸出重定向至文件

self.fileObj=open(file,'a+',1)#改為行緩沖

sys.stdout=self.fileObj

deftoMute(self):#抑制輸出

self.nulObj=open(os.devnull,'w')

sys.stdout=self.nulObj

defrestore(self):

self.content=''

ifself.memObj.closed!=True:

self.memObj.close()

ifself.fileObj.closed!=True:

self.fileObj.close()

ifself.nulObj.closed!=True:

self.nulObj.close()

sys.stdout=self.savedStdout#sys.__stdout__

注意，toFile()方法中，open(name[,mode[,buffering]])調(diào)用選擇行緩沖(無緩沖會(huì)影響性能)。這是為了觀察中間寫入過程，否則只有調(diào)用close()或flush()后輸出才會(huì)寫入文件。內(nèi)部調(diào)用open()方法的缺點(diǎn)是不便于用戶定制寫文件規(guī)則，如模式(覆蓋或追加)和緩沖(行或全緩沖)。

重定向效果如下：

redirObj=RedirectStdout()

sys.stdout=redirObj#本句會(huì)抑制"Let'sbegin!"輸出

print"Let'sbegin!"

#屏顯'HelloWorld!'和'Iamxywang.'(兩行)

redirObj.toCons();print'HelloWorld!';print'Iamxywang.'

#寫入'Howareyou?'和"Can'tcomplain."(兩行)

redirObj.toFile();print'Howareyou?';print"Can'tcomplain."

redirObj.toCons();print"What'up?"#屏顯

redirObj.toMute();print''#無屏顯或?qū)懭?/p>

os.system('echoNeverredirectme!')#控制臺(tái)屏顯'Neverredirectme!'

redirObj.toMemo();print'Whatapity!'#無屏顯或?qū)懭?/p>

redirObj.toCons();print'Hello?'#屏顯

redirObj.toFile();print"Oh,xywangcan'thearme"#該串寫入文件

redirObj.restore()

print'Popup'#屏顯

可見，執(zhí)行toXXXX()語句后，標(biāo)準(zhǔn)輸出流將被重定向到XXXX。此外，toMute()和toMemo()的效果類似，均可抑制輸出。

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