From e0e65212bc3967deb1f9469800e04fe35627802a Mon Sep 17 00:00:00 2001 From: Thomas Steen Rasmussen Date: Tue, 3 Oct 2017 23:45:22 +0200 Subject: [PATCH] mark orderproductrelations as handed_out=True when Ticket objects have been created --- src/shop/admin.py | 2 +- src/shop/models.py | 10 ++++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/src/shop/admin.py b/src/shop/admin.py index 1eeadcb8..cb600227 100644 --- a/src/shop/admin.py +++ b/src/shop/admin.py @@ -97,7 +97,7 @@ class OrderAdmin(admin.ModelAdmin): def mark_order_as_paid(self, request, queryset): for order in queryset.filter(paid=False): - order.mark_as_paid() + order.mark_as_paid(request) mark_order_as_paid.description = 'Mark order(s) as paid' def mark_order_as_refunded(self, request, queryset): diff --git a/src/shop/models.py b/src/shop/models.py index ab6acfb3..5620b5d6 100644 --- a/src/shop/models.py +++ b/src/shop/models.py @@ -185,11 +185,13 @@ class Order(CreatedUpdatedModel): def get_absolute_url(self): return str(reverse_lazy('shop:order_detail', kwargs={'pk': self.pk})) - def mark_as_paid(self): + def mark_as_paid(self, request): self.paid = True self.open = None for order_product in self.orderproductrelation_set.all(): - if order_product.product.category.name == "Tickets": + # if this is a Ticket product? + if order_product.product.ticket_type: + # create the number of tickets required for _ in range(0, order_product.quantity): ticket = ShopTicket( ticket_type=order_product.product.ticket_type, @@ -197,6 +199,10 @@ class Order(CreatedUpdatedModel): product=order_product.product, ) ticket.save() + messages.success(request, "Created %s tickets of type: %s" % (order_product.quantity, order_product.product.ticket_type.name)) + # and mark the OPR as handed_out=True + order_product.handed_out=True + order_product.save() self.save() def mark_as_refunded(self, request):