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

Question

我正在使用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)

任何人都可以建议吗？

谢谢!

Answer 1

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

请不要手动构建包含值的SQL字符串.您可以(并且应该!)单独传递值,使得不必转义并且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

Answer 4

对我来说，盗贼大师的答案和杰米·布朗的答案都不起作用。对我有用的是两者的结合，我想在这里回答，以便它可以帮助其他人。

我需要做的是：

cursor.execute('SELECT LASTVAL()')
id_of_new_row = cursor.fetchone()[0]

即使在之后，该声明lastid = cursor.fetchone()['lastval']对我也不起作用cursor.execute('SELECT LASTVAL()')。仅此声明id_of_new_row = cursor.fetchone()[0]也不起作用。

也许我错过了一些东西。