Add request retry mechanism (unit-finance#143)

* added backoff to requests

* remove prints

* retries default is 1 now

* retries default is 0 now

* changed init method, added retries param

* remove print

* split to functions according to code
and added idempotency_key_is_not_present check

* remove MAX_DURATION

* pr comment

* - idempotency_key_is_present
- added post_create function
for create requests with idempotency_key
- backoff_idempotency_key_handler

* update README.md

* update README.md

Author: axshani

README.md

@@ -56,12 +56,28 @@ Fetching a customer
 ```python
     import os
     from unit import Unit
-    from unit.models.customer import *
 
     token = os.environ.get("token")
     api_url = os.environ.get("api_url")
 
     unit = Unit(api_url, token)
     customer = unit.customers.list().data[0]
     print(customer.id)
+```
+
+## Retrying API Requests
+API requests can fail for many reasons, from network components failures, API rate limits, timeouts or service incidents.
+<br> Create requests without idempotency key won't trigger the retry mechanism, so we recommend to pass an idempotency key where applicable. 
+
+You can read about retries here: https://docs.unit.co/#retries. <br>
+
+the default amount of retries is 0. <br>Unit initialization with retries:
+```python
+    import os
+    from unit import Unit
+
+    token = os.environ.get("token")
+    api_url = os.environ.get("api_url")
+
+    unit = Unit(api_url, token, retries=3)
 ```

e2e_tests/account_end_of_day_test.py

@@ -12,4 +12,3 @@ def test_list_account_end_of_day():
     for a in response.data:
         assert a.type == "accountEndOfDay"
 
-test_list_account_end_of_day()

e2e_tests/application_test.py

@@ -1,5 +1,5 @@
 import os
-import unittest
+import uuid
 from datetime import timedelta
 from unit import Unit
 from unit.models.application import *
@@ -9,6 +9,7 @@
 
 ApplicationTypes = ["individualApplication", "businessApplication", "trustApplication"]
 
+
 def create_individual_application():
     device_fingerprint = DeviceFingerprint.from_json_api({
         "provider": "iovation",
@@ -21,6 +22,7 @@ def create_individual_application():
         Phone("1", "2025550108"),
         ssn="000000003",
         device_fingerprints=[device_fingerprint],
+        idempotency_key=str(uuid.uuid1()),
         jwt_subject="eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9fQ"
     )
 
@@ -73,3 +75,4 @@ def test_update_business_application():
     updated = client.applications.update(PatchApplicationRequest(app.data.id, "businessApplication",
                                                                       tags={"patch": "test-patch"}))
     assert updated.data.type == "businessApplication"
+

e2e_tests/event_test.py

@@ -22,4 +22,3 @@ def test_fire_event():
     response = client.events.fire(event_id)
     assert response.data == []
 
-

e2e_tests/payment_test.py

@@ -22,6 +22,7 @@ def create_book_payment():
 
 def test_list_and_get_payments():
     payments_ids = []
+
     response = client.payments.list()
 
     for t in response.data:

requirements-dev.txt

@@ -1,2 +1,3 @@
 requests==2.26.0
 pytest
+backoff

requirements.txt

@@ -0,0 +1,3 @@
+requests~=2.28.0
+backoff~=2.1.2c
+pytest

setup.py

@@ -12,7 +12,7 @@
     keywords=['unit', 'finance', 'banking',
               'banking-as-a-service', 'API', 'SDK'],
     install_requires=[
-        'requests'
+        'requests', 'backoff'
     ],
     classifiers=[
         'Development Status :: 4 - Beta',

unit/__init__.py

@@ -27,28 +27,28 @@
 
 
 class Unit(object):
-    def __init__(self, api_url, token):
-        self.applications = ApplicationResource(api_url, token)
-        self.customers = CustomerResource(api_url, token)
-        self.accounts = AccountResource(api_url, token)
-        self.cards = CardResource(api_url, token)
-        self.transactions = TransactionResource(api_url, token)
-        self.payments = PaymentResource(api_url, token)
-        self.statements = StatementResource(api_url, token)
-        self.customerTokens = CustomerTokenResource(api_url, token)
-        self.counterparty = CounterpartyResource(api_url, token)
-        self.returnAch = ReturnAchResource(api_url, token)
-        self.applicationForms = ApplicationFormResource(api_url, token)
-        self.fees = FeeResource(api_url, token)
-        self.events = EventResource(api_url, token)
-        self.webhooks = WebhookResource(api_url, token)
-        self.institutions = InstitutionResource(api_url, token)
-        self.atmLocations = AtmLocationResource(api_url, token)
-        self.billPays = BillPayResource(api_url, token)
-        self.api_tokens = APITokenResource(api_url, token)
-        self.authorizations = AuthorizationResource(api_url, token)
-        self.authorization_requests = AuthorizationRequestResource(api_url, token)
-        self.account_end_of_day = AccountEndOfDayResource(api_url, token)
-        self.checkDeposits = CheckDepositResource(api_url, token)
-        self.disputes = DisputeResource(api_url, token)
-        self.rewards = RewardResource(api_url, token)
+    def __init__(self, api_url, token, retries=0):
+        self.applications = ApplicationResource(api_url, token, retries)
+        self.customers = CustomerResource(api_url, token, retries)
+        self.accounts = AccountResource(api_url, token, retries)
+        self.cards = CardResource(api_url, token, retries)
+        self.transactions = TransactionResource(api_url, token, retries)
+        self.payments = PaymentResource(api_url, token, retries)
+        self.statements = StatementResource(api_url, token, retries)
+        self.customerTokens = CustomerTokenResource(api_url, token, retries)
+        self.counterparty = CounterpartyResource(api_url, token, retries)
+        self.returnAch = ReturnAchResource(api_url, token, retries)
+        self.applicationForms = ApplicationFormResource(api_url, token, retries)
+        self.fees = FeeResource(api_url, token, retries)
+        self.events = EventResource(api_url, token, retries)
+        self.webhooks = WebhookResource(api_url, token, retries)
+        self.institutions = InstitutionResource(api_url, token, retries)
+        self.atmLocations = AtmLocationResource(api_url, token, retries)
+        self.billPays = BillPayResource(api_url, token, retries)
+        self.api_tokens = APITokenResource(api_url, token, retries)
+        self.authorizations = AuthorizationResource(api_url, token, retries)
+        self.authorization_requests = AuthorizationRequestResource(api_url, token, retries)
+        self.account_end_of_day = AccountEndOfDayResource(api_url, token, retries)
+        self.checkDeposits = CheckDepositResource(api_url, token, retries)
+        self.disputes = DisputeResource(api_url, token, retries)
+        self.rewards = RewardResource(api_url, token, retries)

unit/api/account_end_of_day_resource.py

@@ -4,8 +4,8 @@
 
 
 class AccountEndOfDayResource(BaseResource):
-    def __init__(self, api_url, token):
-        super().__init__(api_url, token)
+    def __init__(self, api_url, token, retries):
+        super().__init__(api_url, token, retries)
         self.resource = "account-end-of-day"
 
     def list(self, params: ListAccountEndOfDayParams = None) -> Union[UnitResponse[List[AccountEndOfDayDTO]], UnitError]: