如何在Django中让Laptop与Desktop共享递增的InventoryId？

要实现Laptop和Desktop共享连续递增的InventoryId，核心是确保每次创建设备实例时自动生成并关联一个新的InventoryId（利用其AutoField的自增特性），以下是几种推荐的实现方式，均优于信号方案：

1. 抽象基类封装公共逻辑（推荐） #

通过抽象基类将自动创建InventoryId的逻辑统一封装，避免代码重复：

class InventoryId(models.Model):
    inventory_id = models.AutoField(primary_key=True)

class InventoryItem(models.Model):
    # 设置editable=False禁止手动修改，确保自动生成
    inventory_id = models.ForeignKey(InventoryId, on_delete=models.CASCADE, editable=False)

    class Meta:
        abstract = True

    def save(self, *args, **kwargs):
        # 仅在创建新实例时生成InventoryId
        if not self.pk:
            self.inventory_id = InventoryId.objects.create()
        super().save(*args, **kwargs)

class Laptop(InventoryItem):
    name = models.CharField(max_length=30)

class Desktop(InventoryItem):
    name = models.CharField(max_length=30)

2. 重写模型save方法 #

如果不需要复用逻辑，可直接在单个模型中重写save方法：

class InventoryId(models.Model):
    inventory_id = models.AutoField(primary_key=True)

class Laptop(models.Model):
    name = models.CharField(max_length=30)
    inventory_id = models.ForeignKey(InventoryId, on_delete=models.CASCADE, editable=False)

    def save(self, *args, **kwargs):
        if not self.pk:
            self.inventory_id = InventoryId.objects.create()
        super().save(*args, **kwargs)

class Desktop(models.Model):
    name = models.CharField(max_length=30)
    inventory_id = models.ForeignKey(InventoryId, on_delete=models.CASCADE, editable=False)

    def save(self, *args, **kwargs):
        if not self.pk:
            self.inventory_id = InventoryId.objects.create()
        super().save(*args, **kwargs)

3. 自定义模型管理器 #

通过自定义管理器控制实例创建逻辑，适合统一管理创建流程：

class InventoryId(models.Model):
    inventory_id = models.AutoField(primary_key=True)

class InventoryItemManager(models.Manager):
    def create(self, **kwargs):
        # 创建实例前先生成InventoryId
        inventory = InventoryId.objects.create()
        kwargs['inventory_id'] = inventory
        return super().create(**kwargs)

class Laptop(models.Model):
    name = models.CharField(max_length=30)
    inventory_id = models.ForeignKey(InventoryId, on_delete=models.CASCADE, editable=False)
    objects = InventoryItemManager()

class Desktop(models.Model):
    name = models.CharField(max_length=30)
    inventory_id = models.ForeignKey(InventoryId, on_delete=models.CASCADE, editable=False)
    objects = InventoryItemManager()

为什么不推荐信号？ #

信号的问题在于：

• 逻辑分散，模型本身的创建逻辑被拆分到信号处理函数中，维护成本高
• 并发场景下可能出现竞态条件，导致InventoryId关联异常
• 不符合Django"fat models, thin views"的设计原则，模型自身应封装核心逻辑

验证效果 #

当你执行以下操作时，会自动生成连续的InventoryId：

Laptop.objects.create(name="laptop1")  # 关联InventoryId=1
Desktop.objects.create(name="desktop1") # 关联InventoryId=2
Laptop.objects.create(name="laptop2")  # 关联InventoryId=3

内容的提问来源于stack exchange，提问作者Drakan