print>>重定向
這種方式基于print語句的擴展形式,即"printobj>>expr"。其中,obj為一個file-like(尤其是提供write方法的)對象,為None時對應標準輸出(sys.stdout)。expr將被輸出到該文件對象中。示例如下:
memo=cStringIO.StringIO();serr=sys.stderr;file=open('out.txt','w+')
print>>memo,'StringIO';print>>serr,'stderr';print>>file,'file'
print>>None,memo.getvalue()
上述代碼執行后,屏顯為"serr"和"StringIO"(兩行,注意順序),out.txt文件內寫入"file"。
可見,這種方式非常靈活和方便。缺點是不適用于輸出語句較多的場景。
sys.stdout重定向
將一個可寫對象(如file-like對象)賦給sys.stdout,可使隨后的print語句輸出至該對象。重定向結束后,應將sys.stdout恢復最初的缺省值,即標準輸出。
簡單示例如下:
importsys
savedStdout=sys.stdout#保存標準輸出流
withopen('out.txt','w+')asfile:
sys.stdout=file#標準輸出重定向至文件
print'Thismessageisforfile!'
sys.stdout=savedStdout#恢復標準輸出流
print'Thismessageisforscreen!'
注意,IDLE中sys.stdout初值為PseudoOutputFile對象,與sys.__stdout__并不相同。為求通用,本例另行定義變量(savedStdout)保存sys.stdout,下文也將作此處理。此外,本例不適用于經由fromsysimportstdout導入的stdout對象。
以下將自定義多種具有write()方法的file-like對象,以滿足不同需求:
classRedirectStdout:#importos,sys,cStringIO
def__init__(self):
self.content=''
self.savedStdout=sys.stdout
self.memObj,self.fileObj,self.nulObj=None,None,None
#外部的print語句將執行本write()方法,并由當前sys.stdout輸出
defwrite(self,outStr):
#self.content.append(outStr)
self.content+=outStr
deftoCons(self):#標準輸出重定向至控制臺
sys.stdout=self.savedStdout#sys.__stdout__
deftoMemo(self):#標準輸出重定向至內存
self.memObj=cStringIO.StringIO()
sys.stdout=self.memObj
deftoFile(self,file='out.txt'):#標準輸出重定向至文件
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]])調用選擇行緩沖(無緩沖會影響性能)。這是為了觀察中間寫入過程,否則只有調用close()或flush()后輸出才會寫入文件。內部調用open()方法的缺點是不便于用戶定制寫文件規則,如模式(覆蓋或追加)和緩沖(行或全緩沖)。
重定向效果如下:
redirObj=RedirectStdout()
sys.stdout=redirObj#本句會抑制"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''#無屏顯或寫入
os.system('echoNeverredirectme!')#控制臺屏顯'Neverredirectme!'
redirObj.toMemo();print'Whatapity!'#無屏顯或寫入
redirObj.toCons();print'Hello?'#屏顯
redirObj.toFile();print"Oh,xywangcan'thearme"#該串寫入文件
redirObj.restore()
print'Popup'#屏顯
可見,執行toXXXX()語句后,標準輸出流將被重定向到XXXX。此外,toMute()和toMemo()的效果類似,均可抑制輸出。
以上內容為大家介紹了Python重定向之print>>和sys.stdout,希望對大家有所幫助,如果想要了解更多Python相關知識,請關注IT培訓機構:千鋒教育。