如何在Django模型中重写id字段实现自定义格式唯一ID

核心结论 #

不需要重写models.AutoField——AutoField是基于整数的自增字段，无法直接生成包含字母的自定义格式标识。推荐采用两种方案实现需求：数据库存自增主键+动态生成自定义格式ID，或直接存储自定义格式ID并保证唯一性。

方案一：自增主键+动态生成展示用ID #

这种方案保留Django原生自增主键（作为数据库内部唯一标识），通过模型属性生成符合user/0123456ABC格式的对外标识，适合仅需展示或对外提供该格式ID的场景。

代码实现 #

import string
import random
from django.db import models

class User(models.Model):
    # 保留原生自增主键，保障ORM兼容性与数据库性能
    id = models.AutoField(primary_key=True)

    @property
    def custom_id(self):
        # 生成7位数字前缀（不足补0）
        num_part = str(self.id).zfill(7)
        # 从指定字母（示例为A-F）中生成3位后缀
        allowed_chars = string.ascii_uppercase[:6]
        char_part = ''.join(random.choice(allowed_chars) for _ in range(3))
        # 拼接为最终格式
        return f"user/{num_part}{char_part}"

说明 #

• custom_id是动态计算的属性，不会存储在数据库中，每次访问时自动生成
• 可按需调整数字前缀长度、允许的字母范围和后缀位数

方案二：直接存储自定义格式ID作为主键 #

如果需要将自定义格式ID直接作为数据库主键存储，需通过模型方法保证其唯一性与格式正确性。

代码实现 #

import string
import random
from django.db import models
from django.db.utils import IntegrityError

class User(models.Model):
    # 自定义格式ID作为主键，禁止手动编辑
    id = models.CharField(primary_key=True, max_length=15, editable=False)

    def generate_unique_id(self):
        # 允许使用的字符：数字+指定大写字母（示例为A-F）
        allowed_chars = string.digits + string.ascii_uppercase[:6]
        # 循环生成直到得到唯一ID
        while True:
            # 固定开头为0，后续9位从允许字符中随机选取
            id_part = '0' + ''.join(random.choice(allowed_chars) for _ in range(9))
            full_id = f"user/{id_part}"
            if not User.objects.filter(id=full_id).exists():
                return full_id

    def save(self, *args, **kwargs):
        # 新建对象时自动生成ID
        if not self.id:
            self.id = self.generate_unique_id()
        try:
            super().save(*args, **kwargs)
        except IntegrityError:
            # 极端情况出现重复时重新生成并保存
            self.id = self.generate_unique_id()
            super().save(*args, **kwargs)

说明 #

• 使用CharField作为主键，设置editable=False避免手动修改
• 通过查询数据库确保ID唯一性，捕获IntegrityError处理极端重复场景
• 数据量较大时，可引入缓存或更高效的生成算法优化性能

注意事项 #

• 若基于Django自带的AbstractUser扩展，建议将自定义ID字段命名为custom_user_id，保留原生id作为主键，避免冲突
• 可根据需求灵活调整格式规则，比如修改前缀长度、允许的字母范围等

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