使用python将数据复制到Vertica

Question

我使用python和vertica-python库将数据COPY到Vertica DB中

connection = vertica_python.connect(**conn_info)
vsql_cur = connection.cursor()

with open("/tmp/vertica-test-insert", "rb") as fs:
    vsql_cur.copy( "COPY table FROM STDIN DELIMITER ',' ", fs, buffer_size=65536)
    connection.commit()

它插入数据，但只有5行，尽管文件包含更多行。这是否与数据库设置有关，或者是一些客户端问题？

Answer 1

下面的代码适用于我：

对于JSON

# for json file
with open("D:/SampleCSVFile_2kb/tweets.json", "rb") as fs:
    my_file = fs.read().decode('utf-8')
    cur.copy( "COPY STG.unstruc_data FROM STDIN parser fjsonparser()", my_file)
    connection.commit()

对于CSV

# for csv file
with open("D:/SampleCSVFile_2kb/SampleCSVFile_2kb.csv", "rb") as fs:
    my_file = fs.read().decode('utf-8','ignore')
    cur.copy( "COPY STG.unstruc_data FROM STDIN PARSER FDELIMITEDPARSER (delimiter=',', header='false') ", my_file) # buffer_size=65536
    connection.commit()

Answer 2

很可能会有行被拒绝。假设您使用的是7.x，您可以添加：

[ REJECTED DATA {'path' [ ON nodename ] [, ...] | AS TABLE 'reject_table'} ]

您还可以在复制执行后查询以下内容，以查看结果摘要：

SELECTGET_NUM_ACCEPTED_ROWS(),GET_NUM_REJECTED_ROWS();