Python / postgres / psycopg2：获取刚刚插入的行的ID

我正在使用Python和psycopg2来连接到postgres。

当我插入一行…

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

…我如何获得我刚刚插入的行的ID？ 试：

  hundred = cursor.fetchall() 

返回错误，同时使用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) 

只是返回None。

更新：所以currval （即使使用此命令直接进入postgres作品）：

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

任何人都可以build议吗？

谢谢！

Answer 1

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

请不要手动构build包含值的SQLstring。 您可以（也应该！）分别传递值，使其不必转义和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] 

有关更多详细信息，请参阅psycopg文档： http ://initd.org/psycopg/docs/usage.html#passing-parameters-to-sql-queries

Answer 2

我在这里结束了，因为我有类似的问题，但我们正在使用Postgres-XC，它还不支持RETURNING ID子句。 在这种情况下，您可以使用：

  cursor.execute（'INSERT INTO ........'）
 cursor.execute（'SELECT LASTVAL（）'）
 lastid = cursor.fetchone（）['lastval'] 

以防万一它对任何人都有用！

Answer 3

考虑一个RETURNING子句http://www.postgresql.org/docs/8.3/static/sql-insert.html