Extract voucher usage increase/decrease methods to Voucher manager

1 files changed, 9 insertions, 0 deletions

diff --git a/saleor/discount/models.py b/saleor/discount/models.py
index 3b7ee9ae..7bf7ec29 100644
--- a/saleor/discount/models.py
+++ b/saleor/discount/models.py
@@ -4,6 +4,7 @@ from decimal import Decimal
 
 from django.conf import settings
 from django.db import models
+from django.db.models import F
 from django.utils.translation import pgettext, pgettext_lazy
 from django.utils.encoding import python_2_unicode_compatible, smart_text
 from django_countries import countries
@@ -32,6 +33,14 @@ class VoucherQueryset(models.QuerySet):
         queryset = queryset.filter(start_date__lte=today)
         return queryset
 
+    def increase_usage(self, voucher):
+        voucher.used = F('used') + 1
+        voucher.save(update_fields=['used'])
+
+    def decrease_usage(self, voucher):
+        voucher.used = F('used') - 1
+        voucher.save(update_fields=['used'])
+
 
 @python_2_unicode_compatible
 class Voucher(models.Model):