Skip to content

Commit

Permalink
Generate explicit nested resource class methods (#1050)
Browse files Browse the repository at this point in the history 
* Generate explicit nested resource class methods

* Test nested resources

* Revert "Test nested resources"

This reverts commit 296ba8e.

* Add back decorators
  • Loading branch information
@richardm-stripe
richardm-stripe authored Sep 14, 2023
1 parent d4ca00e commit 57f8b88
Show file tree
Hide file tree
Showing 7 changed files with 843 additions and 56 deletions.
7 changes: 5 additions & 2 deletions stripe/api_resources/abstract/nested_resource_class_methods.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,15 +5,15 @@
from stripe.api_resources.abstract import APIResource


# TODO(major): Remove this. It is no longer used except for "nested_resource_url" and "nested_resource_request",
# which are unnecessary ande deprecated.
def nested_resource_class_methods(
resource, path=None, operations=None, resource_plural=None
):
if resource_plural is None:
resource_plural = "%ss" % resource
if path is None:
path = resource_plural
if operations is None:
raise ValueError("operations list required")

def wrapper(cls):
def nested_resource_url(cls, id, nested_id=None):
Expand Down Expand Up @@ -54,6 +54,9 @@ def nested_resource_request(
cls, resource_request_method, classmethod(nested_resource_request)
)

if operations is None:
return cls

for operation in operations:
if operation == "create":

Expand Down
314 changes: 297 additions & 17 deletions stripe/api_resources/account.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,23 +22,10 @@
from stripe.api_resources.person import Person


@nested_resource_class_methods(
"capability",
operations=["retrieve", "update", "list"],
resource_plural="capabilities",
)
@nested_resource_class_methods(
"external_account",
operations=["create", "retrieve", "update", "delete", "list"],
)
@nested_resource_class_methods(
"login_link",
operations=["create"],
)
@nested_resource_class_methods(
"person",
operations=["create", "retrieve", "update", "delete", "list"],
)
@nested_resource_class_methods("capability")
@nested_resource_class_methods("external_account")
@nested_resource_class_methods("login_link")
@nested_resource_class_methods("person")
class Account(
CreateableAPIResource["Account"],
DeletableAPIResource["Account"],
Expand Down Expand Up @@ -245,3 +232,296 @@ def serialize(self, previous):
params[k] = v.serialize(previous.get(k, None))

return params

@classmethod
def retrieve_capability(
cls,
account,
capability,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"get",
"/v1/accounts/{account}/capabilities/{capability}".format(
account=util.sanitize_id(account),
capability=util.sanitize_id(capability),
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def modify_capability(
cls,
account,
capability,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"post",
"/v1/accounts/{account}/capabilities/{capability}".format(
account=util.sanitize_id(account),
capability=util.sanitize_id(capability),
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def list_capabilities(
cls,
account,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"get",
"/v1/accounts/{account}/capabilities".format(
account=util.sanitize_id(account)
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def create_external_account(
cls,
account,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"post",
"/v1/accounts/{account}/external_accounts".format(
account=util.sanitize_id(account)
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def retrieve_external_account(
cls,
account,
id,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"get",
"/v1/accounts/{account}/external_accounts/{id}".format(
account=util.sanitize_id(account), id=util.sanitize_id(id)
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def modify_external_account(
cls,
account,
id,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"post",
"/v1/accounts/{account}/external_accounts/{id}".format(
account=util.sanitize_id(account), id=util.sanitize_id(id)
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def delete_external_account(
cls,
account,
id,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"delete",
"/v1/accounts/{account}/external_accounts/{id}".format(
account=util.sanitize_id(account), id=util.sanitize_id(id)
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def list_external_accounts(
cls,
account,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"get",
"/v1/accounts/{account}/external_accounts".format(
account=util.sanitize_id(account)
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def create_login_link(
cls,
account,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"post",
"/v1/accounts/{account}/login_links".format(
account=util.sanitize_id(account)
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def create_person(
cls,
account,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"post",
"/v1/accounts/{account}/persons".format(
account=util.sanitize_id(account)
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def retrieve_person(
cls,
account,
person,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"get",
"/v1/accounts/{account}/persons/{person}".format(
account=util.sanitize_id(account),
person=util.sanitize_id(person),
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def modify_person(
cls,
account,
person,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"post",
"/v1/accounts/{account}/persons/{person}".format(
account=util.sanitize_id(account),
person=util.sanitize_id(person),
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def delete_person(
cls,
account,
person,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"delete",
"/v1/accounts/{account}/persons/{person}".format(
account=util.sanitize_id(account),
person=util.sanitize_id(person),
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)

@classmethod
def list_persons(
cls,
account,
api_key=None,
stripe_version=None,
stripe_account=None,
**params
):
return cls._static_request(
"get",
"/v1/accounts/{account}/persons".format(
account=util.sanitize_id(account)
),
api_key=api_key,
stripe_version=stripe_version,
stripe_account=stripe_account,
params=params,
)
Loading

0 comments on commit 57f8b88

Please sign in to comment.