python stdout_如何从Python函数调用中捕获stdout输出？

试试这个上下文管理器：from cStringIO import StringIO

import sys

class Capturing(list):

def __enter__(self):

self._stdout = sys.stdout

sys.stdout = self._stringio = StringIO()

return self

def __exit__(self, *args):

self.extend(self._stringio.getvalue().splitlines())

del self._stringio # free up some memory

sys.stdout = self._stdout

用法：with Capturing() as output:

do_something(my_object)

output现在是包含函数调用打印的行的列表。

高级用途：

可能不明显的是，这可以不止一次地完成，结果相互关联：with Capturing() as output:

print 'hello world'

print 'displays on screen'

with Capturing(output) as output: # note the constructor argument

print 'hello world2'

print 'done'

print 'output:', output

输出：displays on screen

done

output: ['hello world', 'hello world2']