使用psycopg2 copy_from()在Postgresql中复制NULL值

Question

这似乎是一个相当受欢迎的问题,但这里的所有答案都没有帮助我解决问题...我的OS X机器上有一个Postgresql 9.5表:

CREATE TABLE test (col1 TEXT, col2 INT)

以下函数使用psycopg2 copy_from()命令:

def test_copy(conn, curs, data):
    cpy = BytesIO()
    for row in data:
         cpy.write('\t'.join([str(x) for x in row]) + '\n')
    print cpy
    cpy.seek(0)
    curs.copy_from(cpy, 'test')

test_copy(connection, [('a', None), ('b', None)])

并将导致此错误:

ERROR:  invalid input syntax for integer: "None"
CONTEXT:  COPY test, line 1, column col2: "None"
STATEMENT:  COPY test FROM stdin WITH DELIMITER AS '    ' NULL AS '\N'

我也试过curs.copy_from(cpy, 'test', null=''),curs.copy_from(cpy, 'test', null='NULL').任何建议都非常感谢.

Answer 1

好的,经过更多的试验和错误,我找到了解决方案:

copy_from(cpy, 'test', null='None')