Внутренний буферизованный вывод Python

Question

Я хотел бы каким-то образом добавить print() в строку или внутренний буфер вместо stdout в Python.

В идеале, я мог бы напечатать это, а затем вывести строку на консоль так, как если бы она была напечатана на stdout для начала. Что-то вроде:

>>> print("output 1", a) 
>>> print("output 2", a) 
>>> print(a) 
output 1 
output 2 

Если вам интересно, что я делаю это ради быстрого рефакторинга на код, который ранее напечатанной непосредственно на консоль.

Answer 1

StringIO действует как виртуальный файл и может быть использован для этой цели, но вы должны быть осторожны, чтобы получить информацию о поведении вы описали:

>>> import io 
>>> a = io.StringIO('') 
>>> print("output 1", file=a) 
>>> print("output 2", file=a) 
>>> print(a.getvalue()) # notice the extra new line at the end... 
output 1 
output 2 

>>> print(a.getvalue(), end='') # use end='' to suppress extra new line 
output 1 
output 2 
>>> print(a) # print(a) will not yield the same results 
<_io.StringIO object at 0x7fe3a23568b8> 
>>> print("output 3", a) # make sure to use "file=", or it will print str(a)! 
output 3 <_io.StringIO object at 0x7fe3a23568b8> 
>>> 

Answer 2

Что вы ищете является StringIO модулем.

Вы просто использовать его таким образом:

import StringIO 

a= StringIO.StringIO() 
a.write('output 1\n') 
print >>a, 'output 2' 

# Retrieve file contents -- this will be 
# 'output 1\noutput 2\n' 
contents = a.getvalue() 

# Close object and discard memory buffer -- 
# .getvalue() will now raise an exception. 
a.close() 

# will output 'output 1\noutput 2\n' 
print contents 

EDIT: Я не видел ответа Джоша перед отправкой моей. Его синтаксис для Python 3 шахта для старых Python 2.x

Answer 3

В общем, метод print() принимает редко используемый параметр, file, который просто должен быть «как файлы» (по умолчанию это стандартный вывод) , Из документа:

Аргумент файла должен быть объектом с методом write(string); если он отсутствует или None, будет использоваться sys.stdout. Буферизация вывода определяется файлом. Используйте file.flush() для обеспечения, например, немедленного появления на экране.

Вы могли бы обеспечить свою собственную реализацию буферной или использовать StringIO как файл-как, если вы хотите сохранить синтаксис print.