Чтобы ответить на вопрос в целом how to redirect standard output to a variable ?
, сделайте следующее:
from io import StringIO
import sys
result = StringIO()
sys.stdout = result
result_string = result.getvalue()
Если вам нужно сделать это только в какой-то функции, сделайте следующее:
old_stdout = sys.stdout
# your function containing the previous lines
my_function()
sys.stdout = old_stdout