diff --git a/core/apps/wherehouse/models/inventory.py b/core/apps/wherehouse/models/inventory.py index 5624080..96b09e1 100644 --- a/core/apps/wherehouse/models/inventory.py +++ b/core/apps/wherehouse/models/inventory.py @@ -31,6 +31,11 @@ class Inventory(BaseModel): def __str__(self): return f'{self.product} in {self.wherehouse}' + + def save(self, *args, **kwargs): + if self.quantity == 0: + self.is_invalid = True + return super().save() class Meta: verbose_name = _('inventar') diff --git a/core/apps/wherehouse/serializers/invalid_product.py b/core/apps/wherehouse/serializers/invalid_product.py index e476ecb..6477f92 100644 --- a/core/apps/wherehouse/serializers/invalid_product.py +++ b/core/apps/wherehouse/serializers/invalid_product.py @@ -52,7 +52,6 @@ class InvalidProductCreateSerializer(serializers.Serializer): wherehouse=validated_data.get('inventory').wherehouse ) invalid_product.witnesses.set(witnesses_ids) - invalid_product.inventory.is_invalid = True invalid_product.inventory.quantity -= validated_data.get('amount') invalid_product.inventory.save() invalid_product.save()