当前位置：首页 > 文章列表 > 文章 > python教程 > SQLAlchemy插入返回自增ID方法

SQLAlchemy插入返回自增ID方法

2026-01-18 22:45:58 0浏览收藏

“纵有疾风来，人生不言弃”，这句话送给正在学习文章的朋友们，也希望在阅读本文《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

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 对象
stmt = insert(users).values(name="Alice").returning(users.c.id)
result = conn.execute(stmt)
new_id = result.scalar_one()  # 返回单个值（即 id）

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
user = User(name="Bob")
session.add(user)
session.flush()  # 不提交，但已执行 INSERT 并获取 id
print(user.id)  # 此时 id 已被赋值
session.commit()  // 按需提交