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)