PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

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

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

Et s'il vous plaît, ne construisez 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 :

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 la documentation psycopg pour plus de détails :http://initd.org/psycopg/docs/usage.html#passing-parameters-to-sql-queries