本文介绍如何通过重写模型的 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 <p>✅ <strong>关键优化说明：</strong> </p><div class="aritcle_card flexRow artxards">
<div class="artcardd flexRow">
<a class="aritcle_card_img" rel="nofollow" href="/xiazai/skill2691" title="Django"><img
src="https://img.php.cn/upload/skill/000/000/081/178928291071323.jpg" alt="Django" onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
<div class="aritcle_card_info flexColumn">
<a rel="nofollow" href="/xiazai/skill2691" title="Django" class="overflowclass">Django</a>
<p class="overflowclass">避免常见的 Django 错误——QuerySet 评估、N+1 查询、迁移冲突和 ORM 陷阱。</p>
</div>
<a rel="nofollow" href="/xiazai/skill2691" title="Django" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span>
</a>
</div>
</div>
- 使用 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 中实现「自动避重生成」的经典实践。
