Блог

Я пытаюсь создать механизм для перенаправления 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")
  

Вывод файла (tmp.txt ) :-

 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 объект.

Я надеюсь, что это имеет смысл.

Вместо того, чтобы изменять стандартный вывод, я бы использовал модуль ведения журнала для достижения того, чего вы хотите.