如何在Django中按自定义规则为不同国家用户自动生成注册号

实现逻辑说明 #

首先明确注册号各部分取值规则：

• 公司缩写：提前定义为固定常量（比如示例中Stack Overflow对应SO）
• 国家代码：直接读取用户关联的Country模型的country_code字段值
• 用户名首字母组合：取用户first_name和last_name的首字母，统一转为大写后拼接
• 6位顺序号：按用户所属国家独立计数，每个国家的第一个用户序号为000001，后续新增同国家用户依次递增，自动补前导零到6位

代码实现 #

推荐直接重写CustomUser模型的save方法，逻辑简单直接，自带事务处理避免序号重复：

第一步：引入依赖、定义固定常量 #

在你的models.py文件头部添加以下代码，公司缩写替换为实际值：

from django.db import transaction
from django.db.models import Max

# 替换为你的实际公司缩写
COMPANY_ABBREVIATION = "SO"

第二步：重写CustomUser的save方法 #

在CustomUser类中添加如下save方法，原有字段逻辑保持不变：

class CustomUser(AbstractUser):
    # 原有字段保持不变
    # ... 省略原有代码 ...

    def save(self, *args, **kwargs):
        # 仅新创建用户且未生成注册号时执行生成逻辑
        if not self.pk and not self.registration_no:
            # 事务包裹避免并发场景下序号重复
            with transaction.atomic():
                # 1. 查询当前国家下已有用户的最大注册号，提取最后6位序号
                max_reg_no = CustomUser.objects.filter(country=self.country).aggregate(
                    max_no = Max("registration_no")
                )["max_no"]

                if max_reg_no:
                    current_seq = int(max_reg_no[-6:]) + 1
                else:
                    # 该国家第一个用户，序号从1开始
                    current_seq = 1
                # 补前导零到6位
                seq_str = f"{current_seq:06d}"

                # 2. 生成姓名首字母组合
                name_initial = f"{self.first_name[0].upper()}{self.last_name[0].upper()}"

                # 3. 拼接完整注册号
                self.registration_no = f"{COMPANY_ABBREVIATION}-{self.country.country_code}-{name_initial}-{seq_str}"
        # 调用原save方法保存数据
        super().save(*args, **kwargs)

注意事项 #

• 请确保first_name和last_name为必填字段，避免取首字母时报错，可在REQUIRED_FIELDS中加入这两个字段
• 请保证Country模型的country_code字段存储的格式符合你的要求（比如示例中的3位数字001）
• 如果系统并发量很高，建议单独创建国家序号表来维护计数，原子性更高，性能更好：

class CountrySeq(models.Model):
    country = models.OneToOneField(Country, on_delete=models.CASCADE)
    current_seq = models.IntegerField(default=0)

    def get_next_seq(self):
        from django.db.models import F
        # F函数保证递增操作原子性
        self.current_seq = F('current_seq') + 1
        self.save()
        self.refresh_from_db()
        return f"{self.current_seq:06d}"

使用时只需调用对应国家CountrySeq实例的get_next_seq方法即可获取新的序号，不会出现重复问题。 #

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