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


99

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

有人可以建议吗?

谢谢!

Answers:


206
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


12
为了澄清起见,idin RETURNING id应该是串行/主键字段的字段名称。
joshden '16

9
游标fetchone给我“没有要获取的结果”。
Leonid

@Leonid你知道了吗?
艾莉森·S

4
@AlisonS @Leonid我有同样的错误,但RETURNING idINSERT查询末尾添加了针对我的错误。
Banjer '16

可能只是一点点注意,但对每个人都需要提及的重要一点:确保在execute()命令之前仅使用游标.execute()而不是游标.mogrify(),否则(例如,在我的情况下)cursor.fetchone ()不会有任何结果!通过在该命令之前仅使用游标.execute()而不使用“ anything”,您将收到一个ID。
TheHeroOfTime

14

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

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

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


4
只需记住-如果在您之后但在lastval()命令返回序列的当前值之前,有人在数据库中插入一行,则在这样的两个语句中这样做有(非常小)竞争条件的风险。
戴夫·托马斯

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.