如何在Django模型中利用其他字段生成主键

嘿，我来帮你搞定这个需求！要让unique_id（作为主键）基于模型里的其他字段生成，最直接的方式是重写模型的save()方法，在对象保存前动态生成符合规则的唯一ID。下面是具体实现方案：

方案1：基于字段缩写+随机字符生成（高唯一性） #

这种方式结合你提供的字段（餐厅名、经理名、州缩写等）生成ID，同时加入随机字符降低重复概率，还会自动处理重复冲突。

完整模型代码 #

import random
import string
from django.db import models
from django.core.exceptions import ValidationError
from django.db import IntegrityError

# 假设你的STATES选项是这样的，可根据实际调整
STATES = [
    ('CA', 'California'),
    ('NY', 'New York'),
    ('TX', 'Texas'),
    # 其他州选项...
]

def validate_mobile(value):
    # 手机号验证逻辑，保持你原有的实现
    if not value.isdigit() or len(value) != 10:
        raise ValidationError("请输入有效的10位数字手机号")

class Restaurant(models.Model):
    unique_id = models.CharField(max_length=10, primary_key=True)
    restaurant_name = models.CharField(max_length=100)
    manager_name = models.CharField(max_length=50)
    email = models.EmailField(max_length=100, unique=True)
    mobile_no = models.CharField(unique=True, validators=[validate_mobile], max_length=10)
    state = models.CharField(choices=STATES, max_length=2)
    city = models.CharField(max_length=20)
    pincode = models.CharField(max_length=6)  # 补全pincode的字段定义

    def generate_unique_id(self):
        """根据其他字段生成基础ID的逻辑"""
        # 提取餐厅名前3个大写字符（去除空格）
        rest_abbr = self.restaurant_name[:3].upper().replace(' ', '')
        # 提取经理名前2个大写字符（去除空格）
        manager_abbr = self.manager_name[:2].upper().replace(' ', '')
        # 直接使用州的缩写
        state_abbr = self.state.upper()
        # 加入1位随机数字增加唯一性
        random_char = random.choice(string.digits)

        # 组合基础前缀，再补全到10位长度
        base_id = f"{rest_abbr}{manager_abbr}{state_abbr}{random_char}"
        if len(base_id) < 10:
            # 不足10位时补随机大小写字母+数字
            base_id += ''.join(random.choices(string.ascii_uppercase + string.digits, k=10 - len(base_id)))
        elif len(base_id) > 10:
            # 超过10位时截断
            base_id = base_id[:10]
        
        return base_id

    def save(self, *args, **kwargs):
        """重写save方法，自动生成唯一主键"""
        # 只有新建对象（unique_id为空）时才生成ID
        if not self.unique_id:
            while True:
                self.unique_id = self.generate_unique_id()
                try:
                    # 尝试保存，若出现主键重复则重新生成
                    super().save(*args, **kwargs)
                    break
                except IntegrityError:
                    continue
        else:
            # 更新对象时直接调用父类save方法
            super().save(*args, **kwargs)

方案2：基于字段缩写+自增序号生成（有序ID） #

如果需要生成有序的主键（比如RESJOHCA001、RESJOHCA002这种），可以用前缀+自增序号的方式：

修改后的generate_unique_id方法 #

def generate_unique_id(self):
    rest_abbr = self.restaurant_name[:3].upper().replace(' ', '')
    manager_abbr = self.manager_name[:2].upper().replace(' ', '')
    state_abbr = self.state.upper()
    
    # 定义ID前缀
    prefix = f"{rest_abbr}{manager_abbr}{state_abbr}"
    # 查询当前所有同前缀的主键
    existing_ids = Restaurant.objects.filter(unique_id__startswith=prefix).values_list('unique_id', flat=True)
    
    # 找出最大的序号
    max_seq = 0
    for uid in existing_ids:
        try:
            # 提取前缀后的序号部分并转为整数
            seq = int(uid[len(prefix):])
            if seq > max_seq:
                max_seq = seq
        except ValueError:
            # 处理不符合规则的ID（比如手动修改过的）
            continue
    
    # 生成新序号，补零到3位（确保总长度10位）
    new_seq = max_seq + 1
    return f"{prefix}{new_seq:03d}"

注意事项 #

• 并发问题：方案2在高并发场景下可能出现重复序号，建议结合Django的select_for_update或者事务来避免冲突。
• 性能：如果数据量很大，方案2的查询会稍慢，这时候可以考虑用单独的计数器表来存每个前缀的当前序号。
• 唯一性保障：无论哪种方案，捕获IntegrityError都是必要的，能兜底处理极端情况下的重复问题。

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