Бесконечный цикл при переопределении метода сохранения в Django
Я хочу переопределить метод сохранения, чтобы
- update quantity if the cart_item already exists
- 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.