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

Getting the id of the last record inserted for Postgresql SERIAL KEY with Python(使用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()

这篇关于使用Python获取为PostgreSQL序列密钥插入的最后一条记录的ID的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！