实时获取Python的print输出流方法

获取Python的print输出流可以通过重定向标准输出流来实现，具体步骤如下：

步骤一：导入sys模块

为了操作标准输出流，我们需要导入Python的sys模块，代码如下：

import sys

步骤二：重定向stdout

重定向stdout，可以将print输出的内容保存到一个指定的文件或者字符串中，而不是显示在控制台上。重定向的步骤如下：

1. 创建一个内存文件对象，可以将其视为临时的输出缓存。

class CatchStringIO(io.StringIO):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.strings = []

    def write(self, string):
        self.strings.append(string)
        super().write(string)

1. 将sys.stdout的原有值保存下来，方便以后的还原。

stdout_sav = sys.stdout

1. 将sys.stdout赋值为内存文件对象，重定向标准输出流

sys.stdout = CatchStringIO()

步骤三：进行打印输出

在重定向后的stdout输出任何内容都会自动保存到内存文件对象中。因此可以进行例子1中的操作。

print('Hello, world!')
print('你好，世界！')

步骤四：读取重定向后的stdout内容

通过之前创建的内存文件对象，即可读取到重定向后的标准输出流内容，代码如下：

sys.stdout.strings

示例说明1

下方例子展示如何在python代码中，实时获取print输出到控制台的结果。

import sys
class CatchStringIO(io.StringIO):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.strings = []

    def write(self, string):
        self.strings.append(string)
        super().write(string)

stdout_sav = sys.stdout
sys.stdout = CatchStringIO()
for i in range(5):
    print(i, end=' ')
    sys.stdout.flush()
    time.sleep(1)
print('\n', 'Done')

sys.stdout = stdout_sav
print(sys.stdout.strings)

该示例运行时，程序每隔1秒钟打印出0~4这5个数字，在执行完成后，即完成了标准输出流的修改。

示例说明2

在下面这个例子中，我们将Python标准输出重定向到一个本地文件中。每次运行时，打印内容会自动保存到相应的文件中，无需在控制台中查看。

import sys

filename = 'test.txt'
stdout_sav = sys.stdout
sys.stdout = open(filename, 'w')
print('Hello, world!')
print('你好，世界！')
sys.stdout = stdout_sav

该示例执行完成后，程序会在本地文件中生成test.txt，并将print输出的内容保存到其中。

上述攻略所述，即为获取python的print输出流方法的完整步骤。