问题描述
我正在尝试创建一种机制来将print
输出重定向到许多变量。以下代码模拟了我正在寻找的内容:-
import sys
import io
class MultIoUt(object):
def __init__(self,stream_out):
self.terminal = sys.stdout
self.output = stream_out
def write(self,message):
self.terminal.write(message)
self.output.write(message)
def flush(self):
self.terminal.flush()
self.output.flush()
vals = {'a1': io.StringIO(),'a2': io.StringIO(),'a3': io.StringIO()}
for i,val in enumerate(vals):
sys.stdout = MultIoUt(vals[val])
[print(x**i,end=' ') for x in range(11)]
print("\n")
with open('temp.txt','w') as f:
for x in vals:
f.write(f"{x} :-\n")
f.write(vals[x].getvalue())
f.write(f"{'='*50}\n")
a1 :-
1 1 1 1 1 1 1 1 1 1 1
0 1 2 3 4 5 6 7 8 9 10
0 1 4 9 16 25 36 49 64 81 100
==================================================
a2 :-
0 1 2 3 4 5 6 7 8 9 10
0 1 4 9 16 25 36 49 64 81 100
==================================================
a3 :-
0 1 4 9 16 25 36 49 64 81 100
==================================================
我在这里要做的是将输出的各个“部分”重定向到vals - a1,a2,a3
的不同变量,并将所有输出转储到终端。奇怪的是,每个连续变量都包含从该点到结束的数据。有没有一种方法可以避免此问题并将每个节保存在不同的变量中?
解决方法
问题是您正在用对象替换sys.stdout:
sys.stdout = MultiOut(vals[val])
并在对象初始化中,将sys.stdout设置为对象属性
self.terminal = sys.stdout
在第二次迭代中,执行此语句时
self.terminal = sys.stdout
sys.stdout是第一次迭代的替换,其中包括第一个MultiOut
对象。
我希望这是有道理的。
而不是修改stdout,我将使用日志记录模块来实现所需的功能。