diff --git a/dtys/models.py b/dtys/models.py index 78a5385..bb05c07 100644 --- a/dtys/models.py +++ b/dtys/models.py @@ -84,3 +84,12 @@ class DTYSModel(models.Model): self.price_stopped = max(self.price_end, self.get_current_price()) self.save() + + def save(self, *args, **kwargs): + if not self.id: + pass + + ProductVariation.objects.filter(id=self.product.id)\ + .update(has_discount=False) + + super(DTYSModel, self).save(*args, **kwargs) diff --git a/dtys/tasks.py b/dtys/tasks.py index f055a1e..cb93b4a 100644 --- a/dtys/tasks.py +++ b/dtys/tasks.py @@ -7,5 +7,5 @@ app = Celery('tasks', broker='redis://localhost:6379/0') def return_dtys(dtys_id): from .models import DTYSModel - DTYSModel.objects.filter( - id=dtys_id, is_ordered=False).update(is_stopped=False) + DTYSModel.objects.filter(id=dtys_id, is_ordered=False)\ + .update(is_stopped=False, price_stopped=0)