Merge pull request #25 from payjp/nyamanaka/add_balance_statement_url

add balance statement URL

Author: natsuki-yamanaka

payjp/resource.py

@@ -436,4 +436,16 @@ class Term(ListableAPIResource):
 
 
 class Balance(ListableAPIResource):
-    pass
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        object.__setattr__(self, 'statement_urls', self.__statement_urls)
+
+    def __statement_urls(self, *args, **kwargs):
+        return self.__class__.statement_urls(self.get('id'), *args, **kwargs)
+
+    @classmethod
+    def statement_urls(cls, id, api_key=None, payjp_account=None, api_base=None, **params):
+        requestor = api_requestor.APIRequestor(api_key, account=payjp_account, api_base=api_base)
+        url = cls.class_url() + f'/{id}/statement_urls'
+        response, api_key = requestor.request('post', url, params)
+        return convert_to_payjp_object(response, api_key, payjp_account, api_base)

payjp/test/test_resources.py

@@ -1210,6 +1210,21 @@ def test_retrieve_balance(self):
             None
         )
 
+    def test_statement_urls(self):
+        payjp.Balance.statement_urls('ba_xxx')
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/balances/ba_xxx/statement_urls',
+            {},
+        )
+
+        balance = payjp.Balance.retrieve('ba_xxx')
+        balance.statement_urls()
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/balances/ba_xxx/statement_urls',
+            {},
+        )
 
 if __name__ == '__main__':
     unittest.main()

payjp/version.py

@@ -1 +1 @@
-VERSION = '0.4.0'
+VERSION = '0.4.1'

setup.py

@@ -14,7 +14,7 @@
 
 setup(
     name="payjp",
-    version="0.4.0",
+    version="0.4.1",
     description='PAY.JP python bindings',
     author="PAY.JP",
     author_email='support@pay.jp',