Сохраните содержимое выписки в Python

Question

Я использую чей-то код в c++, который вызывается python. У меня есть нет способ получить в исходный код и изменить его. Когда вызывается с python, он что-то печатает, и я хочу присвоить этот вывод переменной. Как я могу это сделать?

def print_something(string):
    print(string)

x = print_something('foo')

print(type(x))

Out[5]: NoneType

В идеале этот код присваивает 'foo' x в виде строки.

Answer 1

Может быть, вы можете перенаправить sys.stdout в буфер?

import sys
from io import StringIO

stdout_backup = sys.stdout
sys.stdout = string_buffer = StringIO()

# this function prints something and I cannot change it:
def my_function():
    print('Hello World')

my_function()  # <-- call the function, sys.stdout is redirected now

sys.stdout = stdout_backup  # restore old sys.stdout

string_buffer.seek(0)
print('The output was:', string_buffer.read())

Печать:

The output was: Hello World