最近项目中遇到一个需求,在Django的model中主键要带有前缀的递增类型主键,比如:exp-1, exp-2...,类似.这样,而且在所有的model中,主键里面递增的数据要唯一,不能有重复。也就是如果有A和B两个model,那么当exp-1在A中使用过之后就不允许在B中再使用。在网上找了一圈没有找到特别好的实现方法,自己写了一个,在这里做个记录。
同步发表于个人站点:http://panzhixiang.cn
我采用的方法其实也很简单:
- 创建一个单独的model,里面只有一个models.AutoField类型的字段,可以确保主键中递增的数字是全局唯一的
- 在实际业务model中定义一个models.CharFiled类型的主键
- 修改save方法,为业务模型的主键加上前缀
下面是示例代码,可以参考
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)