Peewee：获取IntegrityError中违反唯一约束的字段

方法一：捕获并解析数据库异常 #

当唯一约束被违反时，数据库会抛出包含约束关联信息的错误，我们可以捕获这个异常并解析错误内容来定位冲突字段：

以你使用的类Peewee风格ORM为例，代码示例如下：

from peewee import IntegrityError

try:
    # 尝试保存用户记录
    user.save()
except IntegrityError as e:
    error_content = str(e)
    # 根据错误信息里的关键词判断冲突字段
    if 'email' in error_content.lower() or 'user_email' in error_content:
        print("邮箱已存在")
    elif 'username' in error_content.lower() or 'user_username' in error_content:
        print("用户名已存在")
    else:
        print("发生未知的唯一约束冲突")

不同数据库的错误信息格式略有差异，但都会包含约束关联的字段名或约束名称（比如PostgreSQL会显示user_email_key这类约束名，MySQL会直接提示Duplicate entry 'xxx' for key 'email'），你可以根据所用数据库的错误格式调整判断逻辑。

方法二：保存前提前查询检查 #

在执行保存操作前，先分别查询数据库中是否已存在相同的email或username：

# 检查邮箱是否已存在
email_conflict = User.select().where(User.email == user.email).exists()
# 检查用户名是否已存在
username_conflict = User.select().where(User.username == user.username).exists()

if email_conflict:
    print("邮箱已存在")
elif username_conflict:
    print("用户名已存在")
else:
    user.save()

这种方法可以提前给出明确提示，但要注意并发场景下的局限性：如果在你的查询操作和保存操作之间，有其他请求插入了相同的email或username，还是会触发唯一约束异常，所以建议将这种方法作为前置检查，同时保留异常处理作为兜底。

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