custom_method decorator for defining static methods for custom API …

…requests

stripe/api_resources/abstract/__init__.py

@@ -21,6 +21,8 @@
 )
 from stripe.api_resources.abstract.verify_mixin import VerifyMixin
 
+from stripe.api_resources.abstract.custom_method import custom_method
+
 from stripe.api_resources.abstract.nested_resource_class_methods import (
     nested_resource_class_methods,
 )

stripe/api_resources/abstract/custom_method.py

@@ -0,0 +1,43 @@
+from __future__ import absolute_import, division, print_function
+
+from stripe import util
+from stripe.six.moves.urllib.parse import quote_plus
+
+
+def custom_method(name, http_verb, http_path=None):
+    if http_verb not in ["get", "post", "delete"]:
+        raise ValueError(
+            "Invalid http_verb: %s. Must be one of 'get', 'post' or 'delete'"
+            % http_verb
+        )
+    if http_path is None:
+        http_path = name
+
+    def wrapper(cls):
+        def custom_method_request(cls, sid, **params):
+            url = "%s/%s/%s" % (
+                cls.class_url(),
+                quote_plus(util.utf8(sid)),
+                http_path,
+            )
+            return cls._static_request(http_verb, url, **params)
+
+        existing_method = getattr(cls, name, None)
+        if existing_method is None:
+            setattr(cls, name, classmethod(custom_method_request))
+        else:
+            # If a method with the same name we want to use already exists on
+            # the class, we assume it's an instance method. In this case, the
+            # new class method is prefixed with `_cls_`, and the original
+            # instance method is decorated with `util.class_method_variant` so
+            # that the new class method is called when the original method is
+            # called as a class method.
+            setattr(cls, "_cls_" + name, classmethod(custom_method_request))
+            instance_method = util.class_method_variant("_cls_" + name)(
+                existing_method
+            )
+            setattr(cls, name, instance_method)
+
+        return cls
+
+    return wrapper

stripe/api_resources/charge.py

@@ -4,8 +4,10 @@
 from stripe.api_resources.abstract import CreateableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("capture", "post")
 class Charge(
     CreateableAPIResource, ListableAPIResource, UpdateableAPIResource
 ):

tests/api_resources/test_charge.py

@@ -60,6 +60,13 @@ def test_is_capturable(self, request_mock):
         )
         assert isinstance(resource, stripe.Charge)
 
+    def test_can_capture(self, request_mock):
+        resource = stripe.Charge.capture(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "post", "/v1/charges/%s/capture" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.Charge)
+
     def test_can_update_dispute(self, request_mock):
         charge = stripe.Charge.retrieve(TEST_RESOURCE_ID)
         resource = charge.update_dispute()