如何在SQLAlchemy批量插入PostgreSQL时忽略唯一约束重复值？

嘿，这个问题我之前也踩过坑！PostgreSQL的唯一约束冲突确实让人头大，不过SQLAlchemy针对PostgreSQL的insert语句提供了专门的处理方案——用on_conflict_do_nothing方法就能轻松跳过重复值，不会再抛出那个烦人的IntegrityError了。

咱们直接来改你的代码：

原来的代码是这样的：

from sqlalchemy.dialects.postgresql import insert
stmt = insert(User, inline=True)
values = [{"id": 1, "name": "David"}, {"id": 2, "name": "Cris"}]
session.execute(stmt, values)
session.commit()

修改后的代码可以这样写：

from sqlalchemy.dialects.postgresql import insert

# 直接把要插入的数据集传给insert的values方法
stmt = insert(User).values([{"id": 1, "name": "David"}, {"id": 2, "name": "Cris"}])

# 指定要忽略的唯一约束——两种方式选其一就行
# 方式1：直接用约束名称（就是你错误提示里的uq_users_name）
stmt = stmt.on_conflict_do_nothing(constraint='uq_users_name')

# 方式2：如果不知道约束名，用构成唯一约束的字段名（这里是name）
# stmt = stmt.on_conflict_do_nothing(index_elements=['name'])

session.execute(stmt)
session.commit()

简单解释一下： #

• on_conflict_do_nothing是PostgreSQL特有的语法封装，告诉数据库当插入的行触发唯一约束冲突时，直接跳过该行，继续处理其他行。
• 你可以根据自己的情况选择指定约束名称或者约束字段：如果清楚约束的名字（比如错误里的uq_users_name），用constraint参数更准确；如果只知道是哪个字段的唯一约束，用index_elements传字段列表就行。

这样修改之后，再执行批量插入时，重复的name值就会被自动忽略，不会再抛出约束冲突的错误，其他符合条件的行也能正常插入啦。

内容的提问来源于stack exchange，提问作者Logovskii Dmitrii