From 600b5497041453964961d6cab0017524a8fa44bb Mon Sep 17 00:00:00 2001
From: PCoder <purple.coder@yahoo.co.uk>
Date: Sat, 21 Oct 2017 20:39:00 +0200
Subject: [PATCH] Handle case where a Stripe account may have been deleted via
dashboard
---
membership/models.py | 21 +++++++++++++++------
1 file changed, 15 insertions(+), 6 deletions(-)
diff --git a/membership/models.py b/membership/models.py
index 73804008..ca403347 100644
--- a/membership/models.py
+++ b/membership/models.py
@@ -210,21 +210,30 @@ class StripeCustomer(models.Model):
# check if user is not in stripe but in database
customer = stripe_utils.check_customer(stripe_customer.stripe_id,
stripe_customer.user, token)
-
+ if "deleted" in customer and customer["deleted"]:
+ raise StripeCustomer.DoesNotExist()
if not customer.sources.data:
stripe_utils.update_customer_token(customer, token)
return stripe_customer
-
except StripeCustomer.DoesNotExist:
user = CustomUser.objects.get(email=email)
stripe_utils = StripeUtils()
stripe_data = stripe_utils.create_customer(token, email, user.name)
if stripe_data.get('response_object'):
stripe_cus_id = stripe_data.get('response_object').get('id')
-
- stripe_customer = StripeCustomer.objects. \
- create(user=user, stripe_id=stripe_cus_id)
-
+ if user.stripecustomer is None:
+ # The user never had an associated Stripe account
+ # So, create one
+ stripe_customer = StripeCustomer.objects.create(
+ user=user, stripe_id=stripe_cus_id
+ )
+ else:
+ # User already had a Stripe account and we are here
+ # because the account was deleted in dashboard
+ # So, we simply update the stripe_id
+ user.stripecustomer.stripe_id = stripe_cus_id
+ user.stripecustomer.save()
+ stripe_customer = user.stripecustomer
return stripe_customer
else:
return None