0

0

sqlalchemy 如何写“插入或更新”并返回 upsert 结果

舞夢輝影

舞夢輝影

发布时间:2026-01-27 19:09:10

|

160人浏览过

|

来源于php中文网

原创

SQLAlchemy 2.0+ 中 PostgreSQL 用 on_conflict_do_update(index_elements=[...]) 实现 upsert,需严格匹配唯一约束字段;SQLite 用 on_conflict_do_replace();MySQL 需手动 prefix_with("ON DUPLICATE KEY UPDATE") 并查 ROW_COUNT();返回完整记录须配合 returning() 且注意数据库版本与 ORM 刷新问题。

sqlalchemy 如何写“插入或更新”并返回 upsert 结果

SQLAlchemy 2.0+ 的 insert().on_conflict_do_update() 怎么用

PostgreSQL 和 SQLite(3.24+)原生支持 upsert,SQLAlchemy 2.0+ 将其封装为 on_conflict_do_update()(PostgreSQL)和 on_conflict_do_replace()(SQLite)。关键不是“有没有”,而是“冲突键怎么写对”——漏掉 index_elements 或写错字段名,会直接报 IntegrityError 而非执行更新。

示例:用户表按 email 唯一约束做 upsert:

from sqlalchemy.dialects.postgresql import insert

stmt = insert(User).values( email="alice@example.com", name="Alice", updated_at=func.now() ) stmt = stmt.on_conflict_do_update( indexelements=["email"], # 必须匹配唯一索引/主键字段,不能写成 ["id"] 除非是主键冲突 set=dict( name=stmt.excluded.name, updated_at=func.now() ) ) session.execute(stmt) session.commit()

  • index_elements 不是表字段任意组合,必须严格对应数据库中已存在的唯一约束或主键定义
  • stmt.excluded 是 PostgreSQL 特有命名,指代本次 INSERT 中被拒绝的那行数据;SQLite 用 excluded 表达相同语义,但需确认驱动版本
  • MySQL 没有原生 ON CONFLICT,得退回到 INSERT ... ON DUPLICATE KEY UPDATE 方式(见下节)

MySQL 怎么写等效的 upsert 并拿到影响行数

MySQL 不支持 on_conflict_do_update(),必须用 prefix_with("ON DUPLICATE KEY UPDATE") 手动拼接。更麻烦的是:SQLAlchemy 默认不暴露 MySQL 的 ROW_COUNT()session.execute() 返回的 Result 对象里没有“到底插入了还是更新了”的明确标识。

实操建议:

  • INSERT 语句末尾加 prefix_with("ON DUPLICATE KEY UPDATE id = LAST_INSERT_ID(id)") ,再调用 connection.exec_driver_sql("SELECT ROW_COUNT()").scalar() 获取实际影响行数(1=插入,2=更新)
  • 更稳的方式是先 SELECT 查是否存在,再分两路处理——虽然多一次查询,但逻辑清晰、可测、不依赖数据库返回值解释
  • 别信 session.merge():它内部是先查后 insert/update,且无法区分操作类型,也不保证原子性

如何让 upsert 返回完整记录(比如新生成的 id 或更新后的值)

PostgreSQL 支持 RETURNING,但必须和 on_conflict_do_update() 配合使用,且只对“最终生效的行”返回——即:插入成功时返回新行,更新成功时返回更新后的行。

Meku
Meku

AI应用和网页开发工具

下载

写法要点:

  • returning() 必须链在 on_conflict_do_update() 之后,不能放在 insert() 后直接调
  • 返回字段要明确写,比如 returning(User.id, User.name, User.updated_at);用 User 类本身会返回整行,但 ORM 映射可能不自动刷新
  • 执行后用 result.fetchone() 拿结果,不是 scalar() ——后者只适合单列

常见坑:returning() 在 SQLite 中仅限 3.35+ 且需启用 sqlite:///...?enable_returning=true 参数,否则静默失败。

ORM 层做 upsert 容易忽略的事务与刷新问题

直接用 session.execute(insert_stmt) 是最可控的,但如果你坚持用 ORM 实例(如 session.merge(user)session.add(user)),要注意:

  • merge() 不触发数据库级唯一冲突检测,它只基于 Python 对象 identity map 做判断,一旦并发写入就可能丢更新
  • add() + flush() 后如果遇到唯一冲突,会抛 IntegrityError,但此时 session 状态已脏,必须 session.rollback() 后重试,不能简单捕获后继续
  • 即使 upsert 成功,ORM 实例的属性也不会自动同步数据库返回值(比如自增 id 或服务端默认值),得手动 session.refresh(instance)

真正需要 upsert 语义时,绕过 ORM 直接写 Core 语句反而更轻、更准、更容易调试。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
mysql修改数据表名
mysql修改数据表名

MySQL修改数据表:1、首先查看数据库中所有的表,代码为:‘SHOW TABLES;’;2、修改表名,代码为:‘ALTER TABLE 旧表名 RENAME [TO] 新表名;’。php中文网还提供MySQL的相关下载、相关课程等内容,供大家免费下载使用。

668

2023.06.20

MySQL创建存储过程
MySQL创建存储过程

存储程序可以分为存储过程和函数,MySQL中创建存储过程和函数使用的语句分别为CREATE PROCEDURE和CREATE FUNCTION。使用CALL语句调用存储过程智能用输出变量返回值。函数可以从语句外调用(通过引用函数名),也能返回标量值。存储过程也可以调用其他存储过程。php中文网还提供MySQL创建存储过程的相关下载、相关课程等内容,供大家免费下载使用。

247

2023.06.21

mongodb和mysql的区别
mongodb和mysql的区别

mongodb和mysql的区别:1、数据模型;2、查询语言;3、扩展性和性能;4、可靠性。本专题为大家提供mongodb和mysql的区别的相关的文章、下载、课程内容,供大家免费下载体验。

281

2023.07.18

mysql密码忘了怎么查看
mysql密码忘了怎么查看

MySQL是一个关系型数据库管理系统,由瑞典MySQL AB 公司开发,属于 Oracle 旗下产品。MySQL 是最流行的关系型数据库管理系统之一,在 WEB 应用方面,MySQL是最好的 RDBMS 应用软件之一。那么mysql密码忘了怎么办呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

515

2023.07.19

mysql创建数据库
mysql创建数据库

MySQL是一个关系型数据库管理系统,由瑞典MySQL AB 公司开发,属于 Oracle 旗下产品。MySQL 是最流行的关系型数据库管理系统之一,在 WEB 应用方面,MySQL是最好的 RDBMS 应用软件之一。那么mysql怎么创建数据库呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

256

2023.07.25

mysql默认事务隔离级别
mysql默认事务隔离级别

MySQL是一种广泛使用的关系型数据库管理系统,它支持事务处理。事务是一组数据库操作,它们作为一个逻辑单元被一起执行。为了保证事务的一致性和隔离性,MySQL提供了不同的事务隔离级别。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

386

2023.08.08

sqlserver和mysql区别
sqlserver和mysql区别

SQL Server和MySQL是两种广泛使用的关系型数据库管理系统。它们具有相似的功能和用途,但在某些方面存在一些显著的区别。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

532

2023.08.11

mysql忘记密码
mysql忘记密码

MySQL是一种关系型数据库管理系统,关系数据库将数据保存在不同的表中,而不是将所有数据放在一个大仓库内,这样就增加了速度并提高了灵活性。那么忘记mysql密码我们该怎么解决呢?php中文网给大家带来了相关的教程以及其他关于mysql的文章,欢迎大家前来学习阅读。

601

2023.08.14

Python 自然语言处理(NLP)基础与实战
Python 自然语言处理(NLP)基础与实战

本专题系统讲解 Python 在自然语言处理(NLP)领域的基础方法与实战应用,涵盖文本预处理(分词、去停用词)、词性标注、命名实体识别、关键词提取、情感分析,以及常用 NLP 库(NLTK、spaCy)的核心用法。通过真实文本案例,帮助学习者掌握 使用 Python 进行文本分析与语言数据处理的完整流程,适用于内容分析、舆情监测与智能文本应用场景。

10

2026.01.27

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
MySQL 教程
MySQL 教程

共48课时 | 1.9万人学习

MySQL 初学入门(mosh老师)
MySQL 初学入门(mosh老师)

共3课时 | 0.3万人学习

简单聊聊mysql8与网络通信
简单聊聊mysql8与网络通信

共1课时 | 811人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号