https://developer.aliyun.com/article/1541306
Python 提供了标准库中的 sys.stdout 对象来代表标准输出。如果我们想将输出重定向到其他位置，例如内存、文件或自定义类，我们可以通过一些技巧来实现。
1. 重定向到内存
要将输出重定向到内存，我们可以使用 StringIO 模块。下面是一个示例：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
| import sys
from io import StringIO
# 创建一个字符串缓冲区来替代默认的标准输出
buffer = StringIO()
sys.stdout = buffer
# 在这里执行需要输出的代码
print("Hello, World!")
print("This is a test.")
# 获取重定向后的内容
output = buffer.getvalue()
# 还原标准输出
sys.stdout = sys.__stdout__
# 打印重定向后的结果
print(output)
|
在上面的示例中，我们首先创建了一个 StringIO 对象 buffer，然后将 sys.stdout 重定向到该对象。接着，我们执行一些输出操作，并使用 getvalue() 方法获取重定向后的内容。最后，我们还原标准输出，并打印出重定向后的结果。
2. 重定向到文件
如果我们想将输出重定向到文件，可以使用 sys.stdout 对象和文件操作符。下面是一个示例：
1
2
3
4
5
6
7
8
9
10
11
12
| import sys
# 打开一个文件以写入输出
with open("output.txt", "w") as f:
# 保存原来的标准输出
original_stdout = sys.stdout
# 重定向标准输出到文件
sys.stdout = f
# 在这里执行需要输出的代码
print("Hello, World!")
print("This is a test.")
# 还原标准输出
sys.stdout = original_stdout
|
在上面的示例中，我们首先打开一个文件，在 “w” 模式下用于写入输出内容。接着，我们保存了原来的标准输出对象 sys.stdout，并将其重定向到文件 f。然后，我们执行一些输出操作，并还原标准输出。
3. 重定向到自定义类
如果我们想将输出重定向到自定义的类中，可以创建一个类，并覆盖该类的 write() 方法，使其接收要输出的内容。下面是一个示例：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
| import sys
import io
class CustomWriter(io.TextIOBase):
def __init__(self):
self.content = ""
def write(self, text):
self.content += text
# 创建一个自定义的输出写入器
custom_writer = CustomWriter()
# 保存原来的标准输出
original_stdout = sys.stdout
# 重定向标准输出到自定义的写入器
sys.stdout = custom_writer
# 在这里执行需要输出的代码
print("Hello, World!")
print("This is a test.")
# 还原标准输出
sys.stdout = original_stdout
# 打印重定向后的结果
print(custom_writer.content)
|
在上面的示例中，我们首先创建了一个自定义类 CustomWriter，需要继承io.TextIOBase。该类具有一个 write() 方法用于接收输出内容。接着，我们创建一个自定义的输出写入器 custom_writer，并将其赋值给 sys.stdout，以将输出重定向到该写入器。然后，我们执行一些输出操作，并最后还原标准输出。
