From c6d8bb0349ccd67b615e99ce9cc07304eeae99ba Mon Sep 17 00:00:00 2001 From: phzhik Date: Thu, 23 Nov 2023 03:39:06 +0400 Subject: [PATCH] * Null-check for gifts --- store/models.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/store/models.py b/store/models.py index 5d392f7..231956b 100644 --- a/store/models.py +++ b/store/models.py @@ -637,16 +637,17 @@ class Checklist(models.Model): self.generate_preview() # Update available gifts count - if self.gift != old_obj.gift: + old_gift = getattr(old_obj, 'gift', None) + if self.gift != old_gift: # Decrement new gift if self.gift: self.gift.available_count = max(0, self.gift.available_count - 1) self.gift.save() # Increment new gift - if old_obj.gift: - old_obj.gift.available_count = max(0, old_obj.gift.available_count + 1) - old_obj.gift.save() + if old_gift: + old_gift.available_count = max(0, old_gift.available_count + 1) + old_gift.save() # Save price details to snapshot if self.price_snapshot_id: