Бесконечный цикл при переопределении метода сохранения в Django

Я хочу переопределить метод сохранения, чтобы

  1. update quantity if the cart_item already exists
  2. insert new entry if the cart_item does not already exists

но каждый раз после метода сохранения он застревает в бесконечном цикле

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")
``

Я думаю, вы можете использовать метод 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```

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() вызывается как для вставки, так и для обновления, и я хотел изменить только поведение вставки, поэтому, чтобы проверить, какая операция выполняется, я воспользовался помощью этой страницы. И если я обновляю модель с помощью метода update(), она не переходит в бесконечный цикл ref.

Вернуться на верх