python - 如何捕获另一个模块的打印输出？

Question

我想知道这在 python 中是否可行:

# module1
def test():
    print('hey')

# module2
import module1

module1.test() # prints to stdout

在不修改 module1 的情况下，有没有办法将其包装在 module2 中，以便我可以捕获print('hey') 在一个变量中？除了将 module1 作为脚本运行之外？

Answer 1

我不想负责修改sys.stdout然后将其恢复到以前的值。以上答案没有任何finally:子句，将其集成到其他重要代码中可能很危险。

https://docs.python.org/3/library/contextlib.html

import contextlib, io

f = io.StringIO()
with contextlib.redirect_stdout(f):
    module1.test()
output = f.getvalue()

您可能需要变量 output这是 <class 'str'>使用重定向的标准输出。

注意:这段代码是从官方文档中提取出来的，只做了一些微不足道的修改(但经过测试)。这个答案的另一个版本已经给出了一个大部分重复的问题:https://stackoverflow.com/a/22434594/1092940

我在这里留下答案，因为它是比 IMO 上的其他解决方案更好的解决方案。