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议吗?

谢谢!

 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

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

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

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