cleanup

codecov · Jan 21, 2025 · 3fd8bd5 · 3fd8bd5
1 parent b569e50
commit 3fd8bd5
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/billing/views.py b/billing/views.py
@@ -107,7 +107,13 @@ def invoice_payment_failed(self, invoice: stripe.Invoice) -> None:
         card = (
             payment_intent.payment_method.card
             if payment_intent.payment_method
-            and not isinstance(payment_intent.payment_method, str)
+            and hasattr(payment_intent.payment_method, 'card')
+            else None
+        )
+        us_bank_account = (
+            payment_intent.payment_method.us_bank_account
+            if payment_intent.payment_method
+            and hasattr(payment_intent.payment_method, 'us_bank_account')
             else None
         )
         template_vars = {
@@ -119,8 +125,8 @@ def invoice_payment_failed(self, invoice: stripe.Invoice) -> None:
             "is_credit_card": True if card else False,
             "card_type": card.brand if card else None,
             # us bank params
-            "is_us_bank": True if payment_intent.payment_method and hasattr(payment_intent.payment_method, 'us_bank_account') else False,
-            "bank_name": payment_intent.payment_method.us_bank_account.bank_name if payment_intent.payment_method and hasattr(payment_intent.payment_method, 'us_bank_account') else None,
+            "is_us_bank": True if us_bank_account else False,
+            "bank_name": us_bank_account.bank_name if us_bank_account else None,
         }
 
         for admin in admins: