本文介绍如何通过重写模型的 save() 方法，在遇到数据库唯一约束冲突（IntegrityError）时自动重新生成唯一值，避免报错中断流程，适用于账户号、邀请码等需强唯一性且需容错生成的场景。
本文介绍如何通过重写模型的 `save()` 方法，在遇到数据库唯一约束冲突（integrityerror）时自动重新生成唯一值，避免报错中断流程，适用于账户号、邀请码等需强唯一性且需容错生成的场景。
在 Django 中，为字段设置 unique=True 可确保数据库层面的唯一性约束，但当插入重复值时会抛出 django.db.IntegrityError，默认行为是中断操作。若业务逻辑要求「不报错，而是自动生成新值」（例如为用户分配唯一账户号），则需主动捕获异常并重试。
以下是一个健壮、可复用的实现方案：
from django.db import IntegrityError
from django.db import models
import random
import string
class Account(models.Model):
user = models.OneToOneField(
'auth.User', # 推荐使用字符串引用避免循环导入
on_delete=models.CASCADE,
related_name='account'
)
account_number = models.CharField(
max_length=25,
unique=True,
help_text="以'217'开头的10位数字账户号"
)
def generate_account_number(self):
"""生成符合要求的唯一账户号：'217' + 10位纯数字"""
digits = ''.join(random.choice(string.digits) for _ in range(10))
return f"217{digits}"
def save(self, *args, **kwargs):
# 若 account_number 为空或未设置，首次生成
if not self.account_number:
self.account_number = self.generate_account_number()
# 最多尝试 10 次，防止无限循环（如表已接近耗尽所有组合）
attempts = 0
max_attempts = 10
while attempts < max_attempts:
try:
super().save(*args, **kwargs)
return # 成功保存，退出
except IntegrityError as e:
if 'account_number' in str(e).lower():
self.account_number = self.generate_account_number()
attempts += 1
else:
raise # 非 account_number 冲突，重新抛出原始异常
raise RuntimeError(
f"无法在 {max_attempts} 次内生成唯一 account_number，请检查字段长度或唯一性设计。"
)✅ 关键优化说明：
- 使用 generate_account_number() 封装逻辑，便于测试与复用；
- 显式捕获 IntegrityError 并仅对 account_number 相关冲突重试，避免掩盖其他数据库错误；
- 设置最大重试次数（如 10 次），防止因高并发或极低熵导致的死循环；
- 若重试失败，抛出清晰的 RuntimeError，便于运维定位问题；
- 字段类型改用 CharField（而非第三方 ShortUUIDField），避免额外依赖与兼容性风险。
⚠️ 注意事项：
- 此方案适用于低至中并发场景；超高并发下建议结合数据库级原子操作（如 PostgreSQL 的 INSERT ... ON CONFLICT DO NOTHING + 应用层轮询），或使用 Redis 分布式锁预生成号段；
- 确保 account_number 字段在数据库索引中为 UNIQUE（Django 迁移会自动创建）；
- 测试时务必覆盖 IntegrityError 路径，可使用 unittest.mock.patch 模拟异常触发重试逻辑。
该方案兼顾简洁性与健壮性，是 Django 中实现「自动避重生成」的经典实践。
