python - Psycopg2 "copy_from"命令，可以忽略引号中的定界符(出现错误)？

Question

我正在尝试使用 copy_from 命令(在 postgres 中使用复制命令的功能)以类似 csv 的结构将数据行加载到 postgres 中。我的数据用逗号分隔(不幸的是，因为我不是数据所有者，所以我不能只更改分隔符)。当我尝试加载一个值在包含逗号的引号中的行时遇到了问题(即，该逗号不应被视为分隔符)。

比如这行数据就可以了:

",Madrid,SN,,SEN,,,SN,173,157"

这行数据不对:

","Dominican, Republic of",MC,,YUO,,,MC,65,162",

部分代码:

    conn = get_psycopg_conn()
    cur = conn.cursor()

    _io_buffer.seek(0) #This buffer is holding the csv-like data
    cur.copy_from(_io_buffer, str(table_name), sep=',', null='', columns=column_names)
    conn.commit()

Answer 1

It looks like copy_from doesn't expose the csv mode or quote options ，这are available form the underlying PostgreSQL COPY command .所以你需要修补 psycopg2 来添加它们，或者 use copy_expert .

我没试过，不过是这样的

curs.copy_expert("""COPY mytable FROM STDIN WITH (FORMAT CSV)""", _io_buffer)

可能就足够了。