使用Python获取为PostgreSQL序列密钥插入的最后一条记录的ID

问题描述

我使用的是不带ORM的SQLAlChemy，即使用手工创建的SQL语句直接与后端数据库交互。在本例中，我使用pg作为后端数据库(作为DB驱动程序)-我不知道这是否会影响答案。

我有这样的语句，为简洁起见，假设conn是到数据库的有效连接：

conn.execute("INSERT INTO user (name, country_id) VALUES ('Homer', 123)")

还假定用户表由列(id [SERIAL PRIMARY KEY], name, country_id)

如何在不再次访问数据库的情况下获取新用户的ID？

推荐答案

您可以这样使用INSERT statement的RETURNING子句：

result = conn.execute("INSERT INTO user (name, country_id) VALUES ('Homer', 123)
                       RETURNING *")

如果您只想要结果id：

result = conn.execute("INSERT INTO user (name, country_id) VALUES ('Homer', 123)
                        RETURNING id")
[new_id] = result.fetchone()