无限循环，同时覆盖 Django 中的保存方法

Question

我想覆盖保存方法，以便

1. 如果 cart_item 已经存在，则更新数量
2. 如果 cart_item 不存在，则插入新条目

但每次保存方法后它都会陷入无限循环

class Cart(models.Model):
    id = models.UUIDField(primary_key=True, default=uuid4)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    def __str__(self) -> str:
        return str(self.id)

class CartItem(models.Model):
    cart = models.ForeignKey(Cart, on_delete=models.CASCADE, related_name='items')
    product = models.ForeignKey(Product, on_delete=models.CASCADE)
    quantity = models.PositiveIntegerField()

    def __str__(self) -> str:
        return self.product.name

    def save(self, *args, **kwargs):
        try:
            cart_item = CartItem.objects.get(cart_id=self.cart.id, product_id=self.product.id)
            cart_item.quantity += self.quantity
            pprint(cart_item)
            cart_item.save() #after this line goes into infinite loop
        except CartItem.DoesNotExist:
            super().save(*args, **kwargs) #after this line goes into infinite loop
            pprint("-------------------------here")
``

Answer 1

我认为你可以使用update_or_create()方法，但如果你想使用你的代码，那么试试这个：

    def save(self, commit=True, *args, **kwargs): ## < --- New parameter here
        old_quantity = 0
        if self.pk: # The object exist if has the pk
            old_quantity = CartItem.objects.get(cart_id=self.cart.id, product_id=self.product.id).quantity

        # If the instance doesn't exists, then it will be created.
        instance = super(CartItem, self).save(commit=False)

        if commit:
            instance.quantity += old_quantity
            instance.save()

        return instance```

Answer 2

def save(self, *args, **kwargs):
    if self._state.adding:
        try:
            cart_item = CartItem.objects.get(cart_id=self.cart.id, product_id=self.product.id)
            quantity = cart_item.quantity + self.quantity              
            CartItem.objects.filter(pk=cart_item.id).update(quantity=quantity)
        except CartItem.DoesNotExist:
            super(CartItem, self).save(*args, **kwargs)
    else:
        super(CartItem, self).save(*args, **kwargs)

使用上面的代码，我能够实现我想要的。 Save() function 被调用来进行插入和更新，我只想改变插入行为，所以为了检查正在执行的操作，我从这个页面获得了帮助。 如果我使用模型的 update() 方法进行更新，它不会 go 进入无限循环ref 。