paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-04 05:53:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

mypy: Replace Any with stripe types in zilencer/lib/stripe.py.

Browse Source
This commit is contained in:
Vishnu Ks
2018-07-03 11:24:39 +00:00
committed by Rishi Gupta
parent b9fa7d7b6d
commit 89fefcd1a2
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
zilencer/lib/stripe.py
View File

@@ -79,21 +79,21 @@ def catch_stripe_errors(func: CallableT) -> CallableT:
return wrapped # type: ignore # https://github.com/python/mypy/issues/1927
@catch_stripe_errors
def get_stripe_customer(stripe_customer_id: int) -> Any:
def get_stripe_customer(stripe_customer_id: int) -> stripe.Customer:
stripe_customer = stripe.Customer.retrieve(stripe_customer_id)
if PRINT_STRIPE_FIXTURE_DATA:
print(''.join(['"retrieve_customer": ', str(stripe_customer), ','])) # nocoverage
return stripe_customer
@catch_stripe_errors
def get_upcoming_invoice(stripe_customer_id: int) -> Any:
def get_upcoming_invoice(stripe_customer_id: int) -> stripe.Invoice:
stripe_invoice = stripe.Invoice.upcoming(customer=stripe_customer_id)
if PRINT_STRIPE_FIXTURE_DATA:
print(''.join(['"upcoming_invoice": ', str(stripe_invoice), ','])) # nocoverage
return stripe_invoice
@catch_stripe_errors
def payment_source(stripe_customer: Any) -> Any:
def payment_source(stripe_customer: stripe.Customer) -> Optional[stripe.Card]:
if stripe_customer.default_source is None:
return None # nocoverage -- no way to get here yet
for source in stripe_customer.sources.data: