推荐方式是ORM插入后调用session.flush()再访问obj.id,兼容所有支持自增主键的数据库;进阶方式用insert().returning()仅限PostgreSQL等支持RETURNING的数据库。
SQLAlchemy 提供了跨数据库兼容的方式,在插入记录后获取自增主键 ID,核心是利用 insert().returning()(较新版本)或更通用的 session.flush() + obj.id 模式。不同数据库对 RETURNING 的支持程度不同,需按需选择策略。
这是最简单、兼容性最好的方法,适用于所有支持自增主键的数据库(如 MySQL、PostgreSQL、SQLite、SQL Server):
session.flush() 触发 SQL 执行但不提交;此时自增 ID 已由数据库生成并回填到对象属性中。RETURNING 语法,完全由 SQLAlchemy 自动适配底层驱动行为。示例:
from sqlalchemy.orm import sessionmaker from mymodels import Useruser = User(name="Alice", email="a@example.com") session.add(user) session.flush() # ✅ 关键:触发 INSERT,ID 已写入 user.id print(user.id) # ✅ 此时已有值(如 123)
后续可继续操作,最后 session.commit()
PostgreSQL 和最新版 SQLite(3.35+)、SQL Server(通过 OUTPUT)原生支持 RETURNING,可一条语句完成插入并取 ID,性能略优。MySQL **不支持** RETURNING,此方式会报错。
from sqlalchemy import insertstmt = insert(User).values(name="Bob", email="b@example.com").returning(User.id) result = session.execute(stmt) new_id = result.scalar_one() # ✅ 返回单个 ID 值
⚠️ 注意:ORM 模型不直接支持 .returning(),需用 Core 语句;且必须确保 dialect 支持(如 postgresql 或 mssql),否则运行时报错。
手动调用底层 cursor 的 lastrowid 不仅破坏抽象,还存在严重兼容问题:
0 或 None;pg_get_serial
_sequence 等函数,无法通用;✅ 正确做法始终走 SQLAlchemy 的 flush / returning / commit 流程,由它协调 dialect 行为。
批量插入(session.bulk_save_objects())默认不填充 ID,因绕过了 ORM 生命周期。若需 ID:
session.add_all() + flush():虽稍慢,但 ID 可用;insert(...).returning() 配合 execute_many(需 SQLAlchemy 2.0+);