Как перенаправить вывод psycopg2 copy_expert() в переменную?

Question

Я хотел бы обработать данные csv таблицы вместо того, чтобы записывать их в файл. Как можно перенаправить psycopg2 copy_expert() в переменную, то вроде этого:

var1 = cur.copy_expert("COPY test TO STDOUT WITH CSV HEADER", sys.stdout) 

Answer 1

Вы можете временно перенаправить sys.stdout строку в Python 3.4.

contextlib.redirect_stdout (NEW_TARGET)

Контекст менеджер для временного перенаправления sys.stdout в другой файл или файл-подобный объект.

Этот инструмент добавляет гибкость существующим функциям или классам, выход связан с выводом на стандартный вывод.

Например, вывод справки() обычно отправляется на sys.stdout. Вы можете захватить этот вывод в строку перенаправив вывод на объект io.StringIO:

f = io.StringIO() 
with redirect_stdout(f): 
    help(pow) 
s = f.getvalue() 

https://docs.python.org/3.4/library/contextlib.html#contextlib.redirect_stdout

Answer 2

Вместо перенаправлять sys.stdout переменной, это легче заменить sys.stdout непосредственно:

text_stream = io.StringIO() 
cur.copy_expert("COPY test TO STDOUT WITH CSV HEADER", text_stream) 
var1 = text_stream.getvalue() 

Это работает, потому что io.StringIO возвращает file object , который copy_expert ожидает второго аргумента.