Handle PaymentMethod type in set_default_card

2020-12-31 15:24:47 +05:30 · 2020-12-31 15:24:47 +05:30 · 41de724904
parent ba92c8e416
commit 41de724904
1 changed files with 5 additions and 1 deletions
--- a/hosting/models.py
+++ b/hosting/models.py
@ -673,7 +673,11 @@ class UserCardDetail(AssignPermissionsMixin, models.Model):
        stripe_utils = StripeUtils()
        cus_response = stripe_utils.get_customer(stripe_api_cus_id)
        cu = cus_response['response_object']
-        cu.default_source = stripe_source_id
+        if stripe_source_id.startswith("pm"):
+            # card is a payment method
+            cu.invoice_settings.default_payment_method = stripe_source_id
+        else:
+            cu.default_source = stripe_source_id
        cu.save()
        UserCardDetail.save_default_card_local(
            stripe_api_cus_id, stripe_source_id