web-dev-qa-db-fra.com

Python / postgres / psycopg2: obtenir l'ID de la ligne qui vient d'être insérée

J'utilise Python et psycopg2 pour interfacer avec postgres.

Quand j'insère une ligne ...

sql_string = "INSERT INTO hundred (name,name_slug,status) VALUES ("
sql_string += hundred_name + ", '" + hundred_slug + "', " + status + ");"
cursor.execute(sql_string)

... comment obtenir l'ID de la ligne que je viens d'insérer? En essayant:

hundred = cursor.fetchall()

renvoie une erreur lors de l'utilisation de RETURNING id:

sql_string = "INSERT INTO domes_hundred (name,name_slug,status) VALUES ("
sql_string += hundred_name + ", '" + hundred_slug + "', " + status + ") RETURNING id;"
hundred = cursor.execute(sql_string)

renvoie simplement None.

MISE À JOUR: currval aussi (même si l'utilisation de cette commande directement dans postgres fonctionne):

sql_string = "SELECT currval(pg_get_serial_sequence('hundred', 'id'));"
hundred_id = cursor.execute(sql_string)

Quelqu'un peut-il conseiller?

merci!

pythonpostgresqlpsycopg2
81
AP257
cursor.execute("INSERT INTO .... RETURNING id")
id_of_new_row = cursor.fetchone()[0]

Et ne créez pas manuellement des chaînes SQL contenant des valeurs. Vous pouvez (et devriez!) Transmettre des valeurs séparément, ce qui rend inutile l'échappement et l'injection SQL impossible:

sql_string = "INSERT INTO domes_hundred (name,name_slug,status) VALUES (%s,%s,%s) RETURNING id;"
cursor.execute(sql_string, (hundred_name, hundred_slug, status))
hundred = cursor.fetchone()[0]

Voir les documents psycopg pour plus de détails: http://initd.org/psycopg/docs/usage.html#passing-parameters-to-sql-queries

165
ThiefMaster

Je me suis retrouvé ici parce que j'avais un problème similaire, mais nous utilisons Postgres-XC, qui ne prend pas encore en charge la clause RETURNING ID. Dans ce cas, vous pouvez utiliser:

cursor.execute ('INSERT INTO ........') 
 cursor.execute ('SELECT LASTVAL ()') 
 lastid = cursor.fetchone () ['lastval']

Juste au cas où c'était utile pour n'importe qui!

11
Jamie Brown

Envisagez une clause RETURNING http://www.postgresql.org/docs/8.3/static/sql-insert.html

2
Mike Graham