PostgreSQL
 sql >> Base de Dados >  >> RDS >> PostgreSQL

obtendo ids de várias linhas inseridas no psycopg2


Você não deveria ser capaz de obter resultados de executemany :

A função é útil principalmente para comandos que atualizam o banco de dados:qualquer conjunto de resultados retornado pela consulta é descartado.

De acordo com os documentos do psycopg2.

Será melhor fazer um loop em um único insert dentro de uma transação ou usando um insert... returning de vários valores , embora no último caso você deva ter cuidado para corresponder os IDs retornados usando outro valor de entrada, você não pode simplesmente assumir que a ordem dos IDs retornados é a mesma que a entrada VALUES Lista.

Quando executo seu teste localmente, ele simplesmente falha:
>>> 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

Testado com psycopg2 2.5.1.