Python:捕获对内存中文件的所有写入

import sys
from io import StringIO

# 创建一个StringIO对象
output = StringIO()

# 将标准输出流重定向到StringIO对象
sys.stdout = output

# 执行需要捕获写入操作的代码
print("Hello, World!")
print("This is a test.")

# 将标准输出流重定向回原来的位置
sys.stdout = sys.__stdout__

# 从StringIO对象中获取写入的内容
output_content = output.getvalue()

# 打印捕获到的写入内容
print(output_content)

Hello, World!
This is a test.