我正在使用python的ftplib编写一个小型FTP客户端,但是包中的某些功能不会返回字符串输出,而是会打印到stdout。我想将stdout重定向到一个我将能够读取输出的对象。

我知道stdout可以使用以下命令重定向到任何常规文件中:

stdout = open("file", "a")


但是我更喜欢不使用本地驱动器的方法。

我正在寻找Java中的诸如BufferedReader之类的东西,该东西可用于将缓冲区包装到流。

评论

我认为stdout = open(“ file”,“ a”)本身不会重定向任何内容。

#1 楼

from cStringIO import StringIO # Python3 use: from io import StringIO
import sys

old_stdout = sys.stdout
sys.stdout = mystdout = StringIO()

# blah blah lots of code ...

sys.stdout = old_stdout

# examine mystdout.getvalue()


评论


+1,您无需保留对原始stdout对象的引用,因为它始终在sys .__ stdout__中可用。请参阅docs.python.org/library/sys.html#sys.__stdout__。

–艾曼·霍里(Ayman Hourieh)
09年8月2日14:00

好吧,这是一个有趣的辩论。绝对原始的stdout可用,但是像这样进行替换时,最好像我一样使用显式保存,因为其他人可能已经替换了stdout,如果您使用stdout,则要破坏它们的替换。

– Ned Batchelder
09年8月2日在14:25

一个线程中的此操作会改变其他线程的行为吗?我的意思是线程安全吗?

– Anuvrat Parashar
2012年9月13日上午11:19

我强烈建议在finally:块中重新分配旧的stdout,因此,如果两者之间出现异常,也要重新分配它。尝试:bkp = sys.stdout ... ...最后:sys.stdout = bkp

– Matthias Kuhn
2014年2月20日下午13:37

如果要在Python 3中使用此功能,请将cStringIO替换为io。

– Anthony Labarre
2014年11月29日在11:41

#2 楼

Python 3.4中有contextlib.redirect_stdout()函数:

import io
from contextlib import redirect_stdout

with io.StringIO() as buf, redirect_stdout(buf):
    print('redirected')
    output = buf.getvalue()


下面的代码示例演示了如何在较旧的Python版本上实现它。

评论


最新的Python上也有redirect_stderr!

– CMCDragonkai
16-10-13在13:19

我认为该解决方案无需添加try / finally块。

–snr
18年7月16日在14:50

#3 楼

只是要添加到Ned的上述答案中:您可以使用它将输出重定向到实现write(str)方法的任何对象。

此方法可以很好地用于“捕获”标准输出。 GUI应用程序。

这是PyQt中一个愚蠢的示例:

import sys
from PyQt4 import QtGui

class OutputWindow(QtGui.QPlainTextEdit):
    def write(self, txt):
        self.appendPlainText(str(txt))

app = QtGui.QApplication(sys.argv)
out = OutputWindow()
sys.stdout=out
out.show()
print "hello world !"


评论


适用于python 2.6和PyQT4。当您无法确定为什么它无效时,对否决工作代码似乎很奇怪!

–尼古拉斯·勒费弗尔(Nicolas Lefebvre)
2011年3月20日15:18

不要忘记也添加flush()!

–将 在13:04标记13年13月

#4 楼

即使存在异常,此方法也将还原sys.stdout。它还在异常之前获取任何输出。

import io
import sys

real_stdout = sys.stdout
fake_stdout = io.BytesIO()   # or perhaps io.StringIO()
try:
    sys.stdout = fake_stdout
    # do what you have to do to create some output
finally:
    sys.stdout = real_stdout
    output_string = fake_stdout.getvalue()
    fake_stdout.close()
    # do what you want with the output_string


使用io.BytesIO()在Python 2.7.10中进行了测试

使用io.StringIO()在Python 3.6.4中进行了测试。


鲍勃,添加到一个案例中,如果您感觉到修改/扩展的代码实验中的任何内容可能在任何意义上都很有趣,否则请随时删除它


广告信息...在寻找一些可行的机制来“抓取”输出时,经过扩展实验的一些评论,由numexpr.print_versions()直接定向到<stdout>(需要清理GUI并将详细信息收集到调试报告中)


# THIS WORKS AS HELL: as Bob Stein proposed years ago:
#  py2 SURPRISEDaBIT:
#
import io
import sys
#
real_stdout = sys.stdout                        #           PUSH <stdout> ( store to REAL_ )
fake_stdout = io.BytesIO()                      #           .DEF FAKE_
try:                                            # FUSED .TRY:
    sys.stdout.flush()                          #           .flush() before
    sys.stdout = fake_stdout                    #           .SET <stdout> to use FAKE_
    # ----------------------------------------- #           +    do what you gotta do to create some output
    print 123456789                             #           + 
    import  numexpr                             #           + 
    QuantFX.numexpr.__version__                 #           + [3] via fake_stdout re-assignment, as was bufferred + "late" deferred .get_value()-read into print, to finally reach -> real_stdout
    QuantFX.numexpr.print_versions()            #           + [4] via fake_stdout re-assignment, as was bufferred + "late" deferred .get_value()-read into print, to finally reach -> real_stdout
    _ = os.system( 'echo os.system() redir-ed' )#           + [1] via real_stdout                                 + "late" deferred .get_value()-read into print, to finally reach -> real_stdout, if not ( _ = )-caught from RET-d "byteswritten" / avoided from being injected int fake_stdout
    _ = os.write(  sys.stderr.fileno(),         #           + [2] via      stderr                                 + "late" deferred .get_value()-read into print, to finally reach -> real_stdout, if not ( _ = )-caught from RET-d "byteswritten" / avoided from being injected int fake_stdout
                       b'os.write()  redir-ed' )#  *OTHERWISE, if via fake_stdout, EXC <_io.BytesIO object at 0x02C0BB10> Traceback (most recent call last):
    # ----------------------------------------- #           ?                              io.UnsupportedOperation: fileno
    #'''                                                    ? YET:        <_io.BytesIO object at 0x02C0BB10> has a .fileno() method listed
    #>>> 'fileno' in dir( sys.stdout )       -> True        ? HAS IT ADVERTISED,
    #>>> pass;            sys.stdout.fileno  -> <built-in method fileno of _io.BytesIO object at 0x02C0BB10>
    #>>> pass;            sys.stdout.fileno()-> Traceback (most recent call last):
    #                                             File "<stdin>", line 1, in <module>
    #                                           io.UnsupportedOperation: fileno
    #                                                       ? BUT REFUSES TO USE IT
    #'''
finally:                                        # == FINALLY:
    sys.stdout.flush()                          #           .flush() before ret'd back REAL_
    sys.stdout = real_stdout                    #           .SET <stdout> to use POP'd REAL_
    sys.stdout.flush()                          #           .flush() after  ret'd back REAL_
    out_string = fake_stdout.getvalue()         #           .GET string           from FAKE_
    fake_stdout.close()                         #                <FD>.close()
    # +++++++++++++++++++++++++++++++++++++     # do what you want with the out_string
    #
    print "\n{0:}\n{1:}{0:}".format( 60 * "/\",# "LATE" deferred print the out_string at the very end reached -> real_stdout
                                     out_string #                   
                                     )
'''
PASS'd:::::
...
os.system() redir-ed
os.write()  redir-ed
/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\
123456789
'2.5'
-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
Numexpr version:   2.5
NumPy version:     1.10.4
Python version:    2.7.13 |Anaconda 4.0.0 (32-bit)| (default, May 11 2017, 14:07:41) [MSC v.1500 32 bit (Intel)]
AMD/Intel CPU?     True
VML available?     True
VML/MKL version:   Intel(R) Math Kernel Library Version 11.3.1 Product Build 20151021 for 32-bit applications
Number of threads used by default: 4 (out of 4 detected cores)
-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\
>>>

EXC'd :::::
...
os.system() redir-ed
/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\
123456789
'2.5'
-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
Numexpr version:   2.5
NumPy version:     1.10.4
Python version:    2.7.13 |Anaconda 4.0.0 (32-bit)| (default, May 11 2017, 14:07:41) [MSC v.1500 32 bit (Intel)]
AMD/Intel CPU?     True
VML available?     True
VML/MKL version:   Intel(R) Math Kernel Library Version 11.3.1 Product Build 20151021 for 32-bit applications
Number of threads used by default: 4 (out of 4 detected cores)
-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\

Traceback (most recent call last):
  File "<stdin>", line 9, in <module>
io.UnsupportedOperation: fileno
'''


#5 楼

从Python 2.6开始,您可以使用任何从io模块中实现TextIOBase API的东西来代替。
此解决方案还使您能够在Python 3中使用sys.stdout.buffer.write()将(已)编码的字节字符串写入stdout(请参阅Python中的stdout)。 3)。
由于无法同时使用StringIOsys.stdout.encoding,因此无法使用sys.stdout.buffer

使用TextIOWrapper的解决方案:

import sys
from io import TextIOWrapper, BytesIO

# setup the environment
old_stdout = sys.stdout
sys.stdout = TextIOWrapper(BytesIO(), sys.stdout.encoding)

# do something that writes to stdout or stdout.buffer

# get output
sys.stdout.seek(0)      # jump to the start
out = sys.stdout.read() # read output

# restore stdout
sys.stdout.close()
sys.stdout = old_stdout


此解决方案适用于Python 2> = 2.6和Python3。

请注意,我们的新版sys.stdout.write()仅接受unicode字符串,而sys.stdout.buffer.write()仅接受字节字符串。
这可能不是旧代码适用于这种情况,但是构建在Python 2和3上运行而无需更改的代码通常适用,这又经常使用sys.stdout.buffer

您可以构建一个可以接受的细微变化write()的unicode和字节字符串:

class StdoutBuffer(TextIOWrapper):
    def write(self, string):
        try:
            return super(StdoutBuffer, self).write(string)
        except TypeError:
            # redirect encoded byte strings directly to buffer
            return super(StdoutBuffer, self).buffer.write(string)


您不必将缓冲区的编码设置为sys.stdout.encoding,但这在使用此方法测试/比较脚本输出时会有所帮助。

评论


设置与Httpie的core.py一起使用的Environment对象的stdout参数时,此答案对我有所帮助。

–fragorl
17年12月6日23:52

#6 楼

python3的上下文管理器:

import sys
from io import StringIO


class RedirectedStdout:
    def __init__(self):
        self._stdout = None
        self._string_io = None

    def __enter__(self):
        self._stdout = sys.stdout
        sys.stdout = self._string_io = StringIO()
        return self

    def __exit__(self, type, value, traceback):
        sys.stdout = self._stdout

    def __str__(self):
        return self._string_io.getvalue()


像这样使用:

>>> with RedirectedStdout() as out:
>>>     print('asdf')
>>>     s = str(out)
>>>     print('bsdf')
>>> print(s, out)
'asdf\n' 'asdf\nbsdf\n'


#7 楼

在Python3.6中,StringIOcStringIO模块不见了,您应该改用io.StringIO,因此您应该像第一个答案那样操作:

import sys
from io import StringIO

old_stdout = sys.stdout
old_stderr = sys.stderr
my_stdout = sys.stdout = StringIO()
my_stderr = sys.stderr = StringIO()

# blah blah lots of code ...

sys.stdout = self.old_stdout
sys.stderr = self.old_stderr

// if you want to see the value of redirect output, be sure the std output is turn back
print(my_stdout.getvalue())
print(my_stderr.getvalue())

my_stdout.close()
my_stderr.close()


评论


您可以通过解释上述代码的工作方式以及对发问者情况的改进来提高答案的质量。

–卡通
17年4月9日在2:28

#8 楼

使用pipe()并写入适当的文件描述符。

https://docs.python.org/library/os.html#file-descriptor-operations

#9 楼

这是另一种看法。记录的contextlib.redirect_stdoutio.StringIO()很棒,但是每天使用仍然有些冗长。这是通过子类化contextlib.redirect_stdout使其成为单线的方法:

 import sys
import io
from contextlib import redirect_stdout

class capture(redirect_stdout):

    def __init__(self):
        self.f = io.StringIO()
        self._new_target = self.f
        self._old_targets = []  # verbatim from parent class

    def __enter__(self):
        self._old_targets.append(getattr(sys, self._stream))  # verbatim from parent class
        setattr(sys, self._stream, self._new_target)  # verbatim from parent class
        return self  # instead of self._new_target in the parent class

    def __repr__(self):
        return self.f.getvalue()  
 


由于__enter__返回,自我,with块退出后,您便可以使用上下文管理器对象。而且,由于__repr__方法,上下文管理器对象的字符串表示实际上是stdout。现在,您有了

 with capture() as message:
    print('Hello World!')
print(str(message)=='Hello World!\n')  # returns True