如何使用 pyscopg2.copy_to 或 copy_expert 写入压缩文件？

Question

我见过的所有例子都使用了psql 和COPY COMMAND | gzip > 'filename' 之类的东西。如果可能的话，我更喜欢使用带有psycopg2 的解决方案，并且我认为将其写出到处理数据然后写出压缩的gzip 文件的字符串缓冲区类型对象可能会很好。

我该怎么做？

Answer 1

psycopg 的文档说copy_to 接受任何类似文件的对象。因此，您可以简单地使用gzip.open 打开一个可写的 gzip 类文件对象：

import gzip

with gzip.open('table-data.gz', 'wb') as gzip_file:
    cursor.copy_to(gzip_file, 'my_table')

或者，如果您更喜欢以某种编码编写文本，并且在 Python 3.3+ 上，您可以使用模式'wt' 并添加encoding='UTF-8' 或类似于gzip.open。