disable has_discount when DTYSModel.save

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)

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)