Django 自定义 id 字段在 Django 模型中

在本文中，我们将介绍如何在 Django 模型中自定义 id 字段。通常情况下，Django 将自动生成一个自增的整数 id 字段用作每个模型的主键。然而，有时我们需要使用不同类型的字段或者自定义生成主键的方式。

使用不同类型的主键字段

除了默认的自增整数类型的 id 字段，Django 也支持使用其他数据类型作为主键。例如，我们可以使用 UUID 字段作为主键。

首先，我们需要导入 UUID 类型：

from django.db import models
import uuid

然后，在我们的模型中使用 UUIDField 类型作为主键字段：

class MyModel(models.Model):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    # 其他字段...

在上面的示例中，我们将 id 字段用 UUIDField 类型替代，并设置为主键字段。默认值使用 uuid.uuid4 函数生成一个随机的唯一标识，并且不可编辑。

自定义生成主键的方式

在某些情况下，我们可能需要根据自定义的规则生成主键。例如，我们希望主键由日期和一个唯一的序列号组成。

为了实现这个需求，我们可以重写模型的 save 方法，在保存对象之前生成自定义主键。

class MyModel(models.Model):
    id = models.CharField(primary_key=True, max_length=20, editable=False)
    # 其他字段...

    def save(self, *args, **kwargs):
        if not self.id:  # 检查是否已有主键值
            today = datetime.now().strftime('%Y%m%d')  # 获取当前日期
            max_id = MyModel.objects.filter(id__startswith=today).count() + 1  # 获取当天已有对象的数量
            self.id = f"{today}-{max_id:02d}"  # 生成新的主键值

        super().save(*args, **kwargs)

在上面的示例中，我们使用 CharField 类型作为主键字段，并设置最大长度为20。在保存对象之前，我们检查是否已经有主键值。如果没有，则获取当前日期和当天已有对象的数量，并生成新的主键值。

示例

假设我们有一个模型用于存储学生信息。我们希望每个学生都有一个唯一的学号，以便于查询和管理。

from django.db import models

class Student(models.Model):
    student_id = models.CharField(primary_key=True, max_length=10)
    name = models.CharField(max_length=20)
    # 其他字段...

在上面的示例中，我们使用 CharField 类型作为主键字段，并设置最大长度为10，以适应学号的位数。

为了确保每个学生都有一个唯一的学号，我们可以在保存学生对象之前检查是否已经存在相同的学号，并在需要时生成一个新的学号。

from django.db import models

def generate_student_id():
    # 生成一个新的学号的逻辑
    pass

class Student(models.Model):
    student_id = models.CharField(primary_key=True, max_length=10, default=generate_student_id, editable=False)
    name = models.CharField(max_length=20)
    # 其他字段...

    def save(self, *args, **kwargs):
        if not self.student_id:  # 检查是否已有学号
            self.student_id = generate_student_id()  # 生成新的学号

        super().save(*args, **kwargs)

在上面的示例中，我们通过调用 generate_student_id() 函数来生成一个新的学号，并设置为默认值。在保存学生对象之前，我们检查是否已有学号值，如果没有，则调用 generate_student_id() 生成一个新的学号。