python - 如何将 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 比将 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 期望第二个参数。