获取一个函数中的print内容
描述
- 假如有一个函数,里面 print 了很多内容,却没 return,现在我们想要获取函数内部 print 的内容,在不修改函数的情况下,我们可以修改系统标准输出到 StringIO,在运行完函数之后再修改回来,那么那个 StringIO 里面的内容就是函数内部的 print 了
代码
import io
import sys
def f():
for i in range(7):
print(i)
if __name__ == '__main__':
sio = io.StringIO()
sys.stdout = sio
f()
sys.stdout = sys.__stdout__ # 将标准输出复原
sio.seek(0)
print(bytes(sio.read(), encoding='utf-8'))
结果:
b'0\n1\n2\n3\n4\n5\n6\n'