获取在psycopg2中插入的多行的ID

Question

我想将psycopg2用于INSERT多行，然后id使用单个查询（按顺序）返回所有s。这就是PostgreSQLRETURNING扩展的目的，并且似乎可以使用正常工作cursor.execute：

cursor.execute(
    "INSERT INTO my_table (field_1, field_2) "
    "VALUES (0, 0), (0, 0) RETURNING id;"
)
print cursor.fetchall()

[(1,), (2,)]

现在，为了传递动态生成的数据，似乎cursor.executemany是一种方法：

data = [(0, 0), (0, 0)]

cursor.executemany(
    "INSERT INTO my_table (field_1, field_2) "
    "VALUES (%s, %s) RETURNING id;",
    data
)

但是，在这种情况下，将cursor.fetchall()产生以下内容：

[(4,), (None,)]

我如何才能正确返回所有ids而不是仅返回一个？

Answer 1

您不应该从executemany以下位置获得结果：

该功能对于更新数据库的命令最有用：查询返回的任何结果集都将被丢弃。

根据psycopg2 docs。

您最好不要insert在事务中循环使用单个值，或者使用多值insert... returning，尽管在后一种情况下，您必须小心使用另一个输入值来匹配返回的ID，您不能仅仅假设返回的ID的顺序与输入VALUES列表相同。

当我在本地运行测试时，它只会失败：

>>> import psycopg2
>>> conn = psycopg2.connect("dbname=regress")
>>> curs = conn.cursor()
>>> curs.execute("create table my_table(id serial primary key, field_1 integer, field_2 integer);")
>>> data = [(0, 0), (0, 0)]
>>> curs.executemany(
...     "INSERT INTO my_table (field_1, field_2) "
...     "VALUES (%s, %s) RETURNING id;",
...     data
... )
>>> 
>>> curs.fetchall()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
psycopg2.ProgrammingError: no results to fetch

用psycopg2 2.5.1测试。