问题:Python PostgreSQL 使用 copy_from to COPY list of objects to table

我正在使用Python 2.7和psycopg2连接到我的数据库服务器(PostgreSQL 9.3),并且我的(产品类)对象列表包含我要插入的项目

products_list = []
products_list.append(product1)
products_list.append(product2)

我想使用copy_from将此产品列表插入到产品表中。我尝试了一些教程,但在将产品列表转换为 CSV 格式时遇到了问题,因为这些值包含单引号、换行符、制表符和双引号。例如(产品说明):

<div class="product_desc">
    Details :
    Product's Name : name
</div>

转义通过在任何单引号之前添加单引号破坏了 HTML 代码,所以我需要使用保存方式将列表转换为 CSV 以复制它?或者使用任何其他方式插入列表而不将其转换为 CSV 格式?

解答

我想通了,首先我创建了一个函数来将我的对象转换为 csv 行

import csv

@staticmethod
def adding_product_to_csv(item, out):
writer = csv.writer(out, quoting=csv.QUOTE_MINIMAL,quotechar='"',delimiter=',',lineterminator="\r\n")
writer.writerow([item.name,item.description])

然后在我的代码中,我使用Python IO创建了一个 csv 文件,将其中的数据存储到COPY中,并使用我之前的函数将每个对象存储在 csv 文件中:

file_name = "/tmp/file.csv"
myfile = open(file_name, 'a')
for item in object_items:
    adding_product_to_csv(item, myfile)

现在我创建了CSV文件,并且可以使用psycopg2中存在的copy_from来复制它:

# For some reason it needs to be closed before copying it to the table
csv_file.close()
cursor.copy_expert("COPY products(name, description) from stdin with delimiter as ',' csv QUOTE '\"' ESCAPE '\"' NULL 'null' ",open(file_name))
conn.commit()
# Clearing the file
open(file_name, 'w').close()

它现在正在工作。