Django中自定义带有前后缀的递增主键

发表于 阅读次数： Waline： 本文字数： 352 阅读时长 ≈ 1 分钟
为Django模型创建带有前缀的唯一递增主键，用简单步骤确保全局唯一。适用于任何模型，提供示例代码供参考，提高项目效率和数据一致性。了解更多实现方法及代码示例。

最近项目中遇到一个需求，在Django的model中主键要带有前缀的递增类型主键，比如：exp-1, exp-2…，类似.这样，而且在所有的model中，主键里面递增的数据要唯一，不能有重复。也就是如果有A和B两个model，那么当exp-1在A中使用过之后就不允许在B中再使用。在网上找了一圈没有找到特别好的实现方法，自己写了一个，在这里做个记录。

我采用的方法其实也很简单：

  1. 创建一个单独的model，里面只有一个models.AutoField类型的字段，可以确保主键中递增的数字是全局唯一的
  2. 在实际业务model中定义一个models.CharFiled类型的主键
  3. 修改save方法，为业务模型的主键加上前缀

下面是示例代码，可以参考

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from django.db import models


class AutoIncrementFields(models.Model):
    """
    provide a unique ID for all models of code sign
    """
    id = models.AutoField(primary_key=True)
    def __str__(self):
        return str(self.id)


class A(models.Model):
    """
    example
    """
    case_id = models.CharField(max_length=128, primary_key=True)  # custom save func will cover this field
    
    def save(self, *args, **kwargs):
        """
        customize primary key like exp-1, exp-2...
        :param args:
        :param kwargs:
        :return:
        """
        _ = AutoIncrementFields.objects.create()
        self.case_id = 'exp-' + str(_.id)
        return super(A, self).save(*args, **kwargs)


class B(models.Model):
    """
    example
    """
    case_id = models.CharField(max_length=128, primary_key=True)  # custom save func will cover this field
    
    def save(self, *args, **kwargs):
        """
        customize primary key like exp-1, exp-2...
        :param args:
        :param kwargs:
        :return:
        """
        _ = AutoIncrementFields.objects.create()
        self.case_id = 'exp-' + str(_.id)
        return super(B, self).save(*args, **kwargs)