使用Python获取Postgresql SERIAL KEY插入的最后一条记录的ID

Question

我使用的是没有 ORM 的 SQLAlchemy，即使用手工编写的 SQL 语句直接与后端数据库交互。在这种情况下，我使用 PG 作为我的后端数据库（psycopg2 作为数据库驱动程序） - 我不知道这是否会影响答案。

到数据库的有效连接：

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

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

我有这样的语句，为了简洁起见，假设 conn是 理想情况下，我可以获取新用户的 ID，而无需再次访问数据库吗？

Answer 1

您也许可以使用 RETURNING 子句>INSERT 语句如下所示：

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()

Answer 2

用户 lastrowid

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

Answer 3

当前的 SQLAlchemy 文档 建议

result.inserted_primary_key 应该可以工作！

Answer 4

Python + SQLAlchemy

提交后，您将在对象中更新主键列 ID（自动增量）。

db.session.add(new_usr)
db.session.commit() #will insert the new_usr data into database AND retrieve id
idd = new_usr.usrID # usrID is the autoincremented primary_key column. 
return jsonify(idd),201 #usrID = 12, correct id from table User in Database.

Answer 5

这个问题在 stackoverflow 上已经被问过很多次了，但我看到的答案都不是全面的。谷歌搜索“sqlalchemy insert get id of new row”会出现很多这样的内容。

SQLAlchemy 分为三个级别。
顶部：ORM。
中：带有表类等的数据库抽象（DBA）。
底部：使用文本函数的 SQL。

对于 OO 程序员来说，ORM 级别看起来很自然，但对于数据库程序员来说，它看起来很难看，而且 ORM 会妨碍。 DBA 层是一个不错的折衷方案。 SQL 层对于数据库程序员来说看起来很自然，而对于纯面向对象的程序员来说则显得陌生。

每个级别都有自己的语法，相似但又不同，足以令人沮丧。除此之外，网上几乎有太多文档，很难找到答案。

我将描述如何在 SQL 层获取我使用的 RDBMS 的插入 id。

Table: User(user_id integer primary autoincrement key, user_name string)
conn: Is a Connection obtained within SQLAlchemy to the DBMS you are using.


SQLite
======
insstmt = text(
    '''INSERT INTO user (user_name)
    VALUES (:usernm) ''' )
# Execute within a transaction (optional)
txn = conn.begin()
result = conn.execute(insstmt, usernm='Jane Doe')
# The id!
recid = result.lastrowid
txn.commit()


MS SQL Server
=============
insstmt = text(
    '''INSERT INTO user (user_name) 
    OUTPUT inserted.record_id
    VALUES (:usernm) ''' )
txn = conn.begin()
result = conn.execute(insstmt, usernm='Jane Doe')
# The id!
recid = result.fetchone()[0]
txn.commit()


MariaDB/MySQL
=============
insstmt = text(
    '''INSERT INTO user (user_name)
    VALUES (:usernm) ''' )
txn = conn.begin()
result = conn.execute(insstmt, usernm='Jane Doe')
# The id!
recid = conn.execute(text('SELECT LAST_INSERT_ID()')).fetchone()[0]
txn.commit()


Postgres
========
insstmt = text(
    '''INSERT INTO user (user_name)
    VALUES (:usernm) 
    RETURNING user_id ''' )
txn = conn.begin()
result = conn.execute(insstmt, usernm='Jane Doe')
# The id!
recid = result.fetchone()[0]
txn.commit()

Answer 6

result.inserted_primary_key

为我工作。唯一需要注意的是，这会返回一个包含该last_insert_id 的列表。

Answer 7

确保使用 fetchrow/fetch 接收返回对象

insert_stmt = user.insert().values(name="homer", country_id="123").returning(user.c.id)

row_id = await conn.fetchrow(insert_stmt)

Answer 8

对于 Postgress 从 python 代码插入很简单，可以在末尾语法的插入语句中使用“RETURNING”关键字和“col_id”（您想要获取最后插入的行 ID 的列的名称）

-

 from sqlalchemy import create_engine
 conn_string = "postgresql://USERNAME:PSWD@HOSTNAME/DATABASE_NAME"
 db = create_engine(conn_string)
 conn = db.connect()
 INSERT INTO emp_table (col_id, Name ,Age) 
        VALUES(3,'xyz',30) RETURNING  col_id;
 or
 (if col_id column is auto increment)
 insert_sql = (INSERT INTO emp_table (Name ,Age) 
        VALUES('xyz',30) RETURNING  col_id;)     
 result = conn.execute(insert_sql)
 [last_row_id] = result.fetchone()
 print(last_row_id)
 #output = 3

前任 -