如何在SQLAlchemy中区分不同的约束错误？

因为IntegrityError是数据库驱动（DBAPI）抛出的原生错误，SQLAlchemy仅做包装处理，所以需要针对具体数据库的错误格式解析，或者通过自定义约束名来统一识别。以下是几种可行方案：

方案1：解析数据库原生错误信息 #

不同数据库的错误提示会包含约束冲突的具体信息，直接提取即可。以你的SQLite场景为例：

from sqlalchemy.orm import DeclarativeBase, MappedAsDataclass, Mapped, mapped_column, Session
from typing import Annotated
from sqlalchemy import create_engine
from sqlalchemy.exc import IntegrityError

database = 'sqlite:///:memory:'
engine = create_engine(database, echo=True)
session = Session(engine)


class Base(MappedAsDataclass, DeclarativeBase):
    pass

class User(Base):
    __tablename__ = 'user'
    id: Mapped[int] = mapped_column(primary_key=True, init=False)
    name: Mapped[Annotated[str, 20]] = mapped_column(unique=True)
    email: Mapped[Annotated[str, 320]]  = mapped_column(unique=True)

if __name__ == '__main__':
    Base.metadata.create_all(engine)

    u1 = User(name='xyz',email='abcd@email.com')
    u2 = User(name='xyz',email='foo@email.com') 
     
    session.add(u1)
    session.add(u2)
    
    try:
        session.commit()
    except IntegrityError as e:
        # 获取数据库原生错误对象
        orig_error = e.orig
        error_msg = str(orig_error)
        
        # SQLite 唯一约束错误解析
        if "UNIQUE constraint failed:" in error_msg:
            _, constraint_info = error_msg.split("UNIQUE constraint failed:", 1)
            table, column = constraint_info.strip().split('.')
            print(f"唯一约束冲突：表 {table} 的 {column} 列值重复")
        
        # 扩展：PostgreSQL 场景的解析逻辑
        elif "duplicate key value violates unique constraint" in error_msg:
            _, constraint_part = error_msg.split('constraint "', 1)
            constraint_name = constraint_part.split('"')[0]
            # 提前映射约束名到列名
            constraint_column_map = {
                "user_name_key": "name",
                "user_email_key": "email"
            }
            print(f"唯一约束冲突：{constraint_column_map.get(constraint_name, constraint_name)} 列值重复")

方案2：自定义约束名，统一识别 #

在定义列时给唯一约束指定明确名称，避免数据库自动生成的随机约束名，方便后续识别：

class User(Base):
    __tablename__ = 'user'
    id: Mapped[int] = mapped_column(primary_key=True, init=False)
    # 自定义唯一约束名
    name: Mapped[Annotated[str, 20]] = mapped_column(unique=True, constraint_name="user_name_unique")
    email: Mapped[Annotated[str, 320]]  = mapped_column(unique=True, constraint_name="user_email_unique")

之后解析错误信息中的约束名，通过预设的映射表就能直接对应到具体列。

方案3：辅助验证（可选） #

如果需要提前预判，可以在提交前主动查询数据库，检查当前数据是否违反唯一约束，但这种方法会增加额外查询，适合数据量小的场景。

内容的提问来源于stack exchange，提问作者Soham Jobanputra