3

我正在使用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 格式？

4

1 回答 1

1

我想通了，首先我创建了一个函数来将我的对象转换为 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])

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

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

现在我创建了CSV文件，并且可以使用以下文件复制copy_from它psycopg2：

# 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()

它现在正在工作。

于 2015-06-17T12:56:55.027 回答