获取一个函数中的print内容

获取一个函数中的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'