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

panzhixiang

发布于 2024-10-30 19:26:50

1050

发布于 2024-10-30 19:26:50

文章被收录于专栏：panzhixiang

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

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

创建一个单独的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)

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021-04-06，如有侵权请联系 cloudcommunity@tencent.com 删除

self