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

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

同步发表于个人站点:http://panzhixiang.cn

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

  1. 创建一个单独的model,里面只有一个models.AutoField类型的字段,可以确保主键中递增的数字是全局唯一的
  2. 在实际业务model中定义一个models.CharFiled类型的主键
  3. 修改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)
上一篇:tp5 exp表达式与raw()方法


下一篇:softmax 在计算图中的前向和后向