From ee186f047e49a604b15da1c99f1a850bfdd9f6a7 Mon Sep 17 00:00:00 2001 From: Nathan Chapman Date: Wed, 11 May 2022 20:39:14 -0600 Subject: [PATCH] Fix when no coupon error --- src/storefront/views.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/storefront/views.py b/src/storefront/views.py index 63da3e6..c309aed 100644 --- a/src/storefront/views.py +++ b/src/storefront/views.py @@ -268,10 +268,13 @@ class OrderCreateView(CreateView): shipping_address = self.request.session.get('shipping_address') form.instance.customer, form.instance.shipping_address = get_or_create_customer(self.request, form, shipping_address) form.instance.status = OrderStatus.DRAFT - coupon = get_object_or_404( - Coupon, - code=self.request.session.get('coupon_code') - ) + try: + coupon = Coupon.objects.get( + code=self.request.session.get('coupon_code') + ) + except ObjectDoesNotExist: + coupon = None + if coupon: form.instance.coupon = coupon coupon.users.add(form.instance.customer)