SQLAlchemy插入返回自增ID方法

“纵有疾风来，人生不言弃”，这句话送给正在学习文章的朋友们，也希望在阅读本文《SQLAlchemy 插入后返回自增 ID 的写法，需根据数据库类型（MySQL 或 PostgreSQL）进行适配。以下是兼容两种数据库的实现方式。✅ 1. 使用 session.add() + session.flush() 返回自增 ID示例代码：from sqlalchemy import create_engine, Column, Integer, String from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import sessionmaker Base = declarative_base() class User(Base): __tablename__ = 'users' id = Column(Integer, primary_key=True) name = Column(String(50)) # 创建引擎和会话 engine = create_engine('mysql+pymysql://user:password@localhost/dbname') # MySQL # engine = create_engine('postgresql+psycopg2://user:password@localhost/dbname') # PostgreSQL Session = sessionmaker(bind=engine) session = Session() # 插入数据并获取自增 ID new_user = User(name="Alice") session.add(new_user) session.flush() # 刷新会话，确保 ID 已生成 print("Inserted ID:", new_user.id) # 适用于 MySQL 和 PostgreSQL session.commit()✅ 2. 使用 insert() 语句 + returning 子句（PostgreSQL）对于 PostgreSQL，可以使用 SQLAlchemy 的 insert() 语句并结合 `》后，能够真的帮助到大家。我也会在后续的文章中，陆续更新文章相关的技术文章，有好的建议欢迎大家在评论留言，非常感谢！

SQLAlchemy 推荐使用 Core 的 insert().returning() 实现跨数据库插入后返回自增 ID，PostgreSQL 原生支持，MySQL 8.0.19+ 配合新驱动也支持；ORM 则通过 session.flush() 后直接读取主键。

SQLAlchemy 提供了统一的写法来实现“插入后返回自增 ID”，无需为 MySQL 和 PostgreSQL 分别写逻辑，关键在于正确使用 insert().returning()（现代 Core）或 ORM 的 session.flush() + 主键属性（传统 ORM），并配合数据库驱动支持。

推荐方式：使用 Core 的 insert().returning()

这是最直接、跨数据库兼容性最好的方法（PostgreSQL 原生支持，MySQL 8.0.19+ 也支持 RETURNING）。只需确保使用较新版本的 SQLAlchemy（≥1.4）和对应驱动（如 pymysql ≥1.0 或 mysqlclient ≥2.1，psycopg2 ≥2.8）：

• PostgreSQL 自动启用 RETURNING
• MySQL 在满足条件（8.0.19+ + 启用 RETURNING 支持的驱动）时也会走原生 RETURNING；否则 SQLAlchemy 会自动回退到 lastrowid 方式（仍能拿到 ID，只是非单条语句）

示例（Core）：

from sqlalchemy import insert, create_engine
from mymodels import users  # 假设是 Table 对象
<p>stmt = insert(users).values(name="Alice").returning(users.c.id)
result = conn.execute(stmt)
new_id = result.scalar_one()  # 返回单个值（即 id）
</p>

ORM 方式：提交前 flush，直接读主键

对 ORM 模型实例，不需要手动写 SQL。只要主键列声明了 primary_key=True 且未显式赋值，SQLAlchemy 会在 session.flush() 后自动填充自增 ID（底层调用 cursor.lastrowid 或 RETURNING）：

• MySQL：依赖驱动的 lastrowid（如 pymysql 的 cursor.lastrowid）
• PostgreSQL：优先用 RETURNING，fallback 到 pg_get_last_oid() 或序列查询

示例（ORM）：

from mymodels import User
<p>user = User(name="Bob")
session.add(user)
session.flush()  # 不提交，但已执行 INSERT 并获取 id
print(user.id)  # 此时 id 已被赋值</p><h1>session.commit()  // 按需提交</h1><p></p>

注意驱动与方言兼容性

某些旧驱动或配置可能导致 RETURNING 不生效，可显式检查行为：

• 确认 MySQL 版本 ≥8.0.19 且连接 URL 中未禁用（如避免加 ?use_returning=false）
• PostgreSQL 推荐用 psycopg2（asyncpg 也支持 RETURNING）
• 若用 SQLite，RETURNING 不可用，但 lastrowid 依然可靠

可通过 engine.dialect.supports_sane_rowcount_returning 查看是否启用原生 RETURNING。

不推荐：手写 SELECT LAST_INSERT_ID() 或 SELECT currval()

这类写法破坏可移植性，且易出错（如事务隔离、多线程竞争、序列名硬编码等）。SQLAlchemy 已封装好适配逻辑，直接信任其 flush() 或 returning() 即可。

今天关于《SQLAlchemy插入返回自增ID方法》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！