amal/dynamicweb
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Use hasattr to check if a user already has a stripecustomer object

Browse source
This commit is contained in:
M.Ravi 2017-10-26 15:00:54 +02:00
commit ef9dc446db
2 changed files with 12 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

14
membership/models.py
View file

@ -221,19 +221,19 @@ class StripeCustomer(models.Model):
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')
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:
if hasattr(user, 'stripecustomer'):
# 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
else:
# The user never had an associated Stripe account
# So, create one
stripe_customer = StripeCustomer.objects.create(
user=user, stripe_id=stripe_cus_id
)
return stripe_customer
else:
return None