From c2023eaaa8cf844bbdb4741ba2150611578ffb75 Mon Sep 17 00:00:00 2001 From: Vishnu Ks Date: Fri, 3 Aug 2018 16:08:16 +0530 Subject: [PATCH] billing: Mention the exact type of stripe.Customers subscriptions attribute. --- stubs/stripe/__init__.pyi | 3 ++- stubs/stripe/api_resources/list_object/__init__.pyi | 10 ++++++++-- zilencer/tests/test_stripe.py | 1 - 3 files changed, 10 insertions(+), 4 deletions(-) diff --git a/stubs/stripe/__init__.pyi b/stubs/stripe/__init__.pyi index 1dd9667e92..85b364b685 100644 --- a/stubs/stripe/__init__.pyi +++ b/stubs/stripe/__init__.pyi @@ -1,5 +1,6 @@ import stripe.error as error import stripe.util as util +from stripe.api_resources.list_object import SubscriptionListObject from typing import Optional, Any, Dict, List @@ -9,7 +10,7 @@ class Customer: default_source: Card created: int id: str - subscriptions: List[Subscription] + subscriptions: SubscriptionListObject @staticmethod def retrieve(customer_id: str, expand: Optional[List[str]]) -> Customer: diff --git a/stubs/stripe/api_resources/list_object/__init__.pyi b/stubs/stripe/api_resources/list_object/__init__.pyi index c35e02b96d..d1e87884a0 100644 --- a/stubs/stripe/api_resources/list_object/__init__.pyi +++ b/stubs/stripe/api_resources/list_object/__init__.pyi @@ -1,2 +1,8 @@ -class ListObject: - ... +from stripe import Subscription +from typing import Optional, Any, Dict, List, Iterator + +class SubscriptionListObject: + data: List[Subscription] + + def __iter__(self) -> Iterator[Subscription]: + ... diff --git a/zilencer/tests/test_stripe.py b/zilencer/tests/test_stripe.py index d78cfd177e..e3348177fb 100644 --- a/zilencer/tests/test_stripe.py +++ b/zilencer/tests/test_stripe.py @@ -6,7 +6,6 @@ import ujson from django.core import signing import stripe -from stripe.api_resources.list_object import ListObject from zerver.lib.actions import do_deactivate_user, do_create_user, \ do_activate_user, do_reactivate_user, activity_change_requires_seat_update