首页 / 知识

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

2023-11-12 13:33:00

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!'#无屏显或写入