added support for idempotency key (#2)

* added support for idempotency key --------- Co-authored-by: Michal Schielmann <[email protected]>
shift4developer · Aug 27, 2024 · 9cccad7 · 9cccad7
1 parent c0158f6
commit 9cccad7
Show file tree

Hide file tree

Showing 23 changed files with 852 additions and 60 deletions.
diff --git a/shift4/blacklist.py b/shift4/blacklist.py
@@ -2,8 +2,8 @@
 
 
 class Blacklist(Resource):
-    def create(self, params):
-        return self._post("/blacklist", params)
+    def create(self, params, request_options=None):
+        return self._post("/blacklist", params, request_options=request_options)
 
     def get(self, blacklist_rule_id):
         return self._get("/blacklist/%s" % blacklist_rule_id)

diff --git a/shift4/cards.py b/shift4/cards.py
@@ -2,14 +2,20 @@
 
 
 class Cards(Resource):
-    def create(self, customer_id, params):
-        return self._post("/customers/%s/cards" % customer_id, params)
+    def create(self, customer_id, params, request_options=None):
+        return self._post(
+            "/customers/%s/cards" % customer_id, params, request_options=request_options
+        )
 
     def get(self, customer_id, card_id):
         return self._get("/customers/%s/cards/%s" % (customer_id, card_id))
 
-    def update(self, customer_id, card_id, params):
-        return self._post("/customers/%s/cards/%s" % (customer_id, card_id), params)
+    def update(self, customer_id, card_id, params, request_options=None):
+        return self._post(
+            "/customers/%s/cards/%s" % (customer_id, card_id),
+            params,
+            request_options=request_options,
+        )
 
     def delete(self, customer_id, card_id):
         return self._delete("/customers/%s/cards/%s" % (customer_id, card_id))

diff --git a/shift4/charges.py b/shift4/charges.py
@@ -2,20 +2,26 @@
 
 
 class Charges(Resource):
-    def create(self, params):
-        return self._post("/charges", params)
+    def create(self, params, request_options=None):
+        return self._post("/charges", params, request_options=request_options)
 
     def get(self, charge_id):
         return self._get("/charges/%s" % charge_id)
 
-    def update(self, charge_id, params):
-        return self._post("/charges/%s" % charge_id, params)
+    def update(self, charge_id, params, request_options=None):
+        return self._post(
+            "/charges/%s" % charge_id, params, request_options=request_options
+        )
 
     def list(self, params=None):
         return self._get("/charges", params)
 
-    def capture(self, charge_id):
-        return self._post("/charges/%s/capture" % charge_id)
+    def capture(self, charge_id, request_options=None):
+        return self._post(
+            "/charges/%s/capture" % charge_id, request_options=request_options
+        )
 
-    def refund(self, charge_id, params=None):
-        return self._post("/charges/%s/refund" % charge_id, params)
+    def refund(self, charge_id, params=None, request_options=None):
+        return self._post(
+            "/charges/%s/refund" % charge_id, params, request_options=request_options
+        )
diff --git a/shift4/credits.py b/shift4/credits.py
@@ -2,14 +2,16 @@
 
 
 class Credits(Resource):
-    def create(self, params):
-        return self._post("/credits", params)
+    def create(self, params, request_options=None):
+        return self._post("/credits", params, request_options=request_options)
 
     def get(self, credit_id):
         return self._get("/credits/%s" % credit_id)
 
-    def update(self, credit_id, params):
-        return self._post("/credits/%s" % credit_id, params)
+    def update(self, credit_id, params, request_options=None):
+        return self._post(
+            "/credits/%s" % credit_id, params, request_options=request_options
+        )
 
     def list(self, params=None):
         return self._get("/credits", params)
diff --git a/shift4/customers.py b/shift4/customers.py
@@ -2,14 +2,16 @@
 
 
 class Customers(Resource):
-    def create(self, params):
-        return self._post("/customers", params)
+    def create(self, params, request_options=None):
+        return self._post("/customers", params, request_options=request_options)
 
     def get(self, customer_id):
         return self._get("/customers/%s" % customer_id)
 
-    def update(self, customer_id, params):
-        return self._post("/customers/%s" % customer_id, params)
+    def update(self, customer_id, params, request_options=None):
+        return self._post(
+            "/customers/%s" % customer_id, params, request_options=request_options
+        )
 
     def delete(self, customer_id):
         return self._delete("/customers/%s" % customer_id)

diff --git a/shift4/disputes.py b/shift4/disputes.py
@@ -5,11 +5,15 @@ class Disputes(Resource):
     def get(self, dispute_id):
         return self._get("/disputes/%s" % dispute_id)
 
-    def update(self, dispute_id, params):
-        return self._post("/disputes/%s" % dispute_id, params)
+    def update(self, dispute_id, params, request_options=None):
+        return self._post(
+            "/disputes/%s" % dispute_id, params, request_options=request_options
+        )
 
-    def close(self, dispute_id):
-        return self._post("/disputes/%s/close" % dispute_id)
+    def close(self, dispute_id, request_options=None):
+        return self._post(
+            "/disputes/%s/close" % dispute_id, request_options=request_options
+        )
 
     def list(self, params=None):
         return self._get("/disputes", params)
diff --git a/shift4/payment_methods.py b/shift4/payment_methods.py
@@ -2,8 +2,8 @@
 
 
 class PaymentMethods(Resource):
-    def create(self, params):
-        return self._post("/payment-methods", params)
+    def create(self, params, request_options=None):
+        return self._post("/payment-methods", params, request_options=request_options)
 
     def get(self, payment_method_id):
         return self._get("/payment-methods/%s" % payment_method_id)

diff --git a/shift4/plans.py b/shift4/plans.py
@@ -2,14 +2,16 @@
 
 
 class Plans(Resource):
-    def create(self, params):
-        return self._post("/plans", params)
+    def create(self, params, request_options=None):
+        return self._post("/plans", params, request_options=request_options)
 
     def get(self, plan_id):
         return self._get("/plans/%s" % plan_id)
 
-    def update(self, plan_id, params):
-        return self._post("/plans/%s" % plan_id, params)
+    def update(self, plan_id, params, request_options=None):
+        return self._post(
+            "/plans/%s" % plan_id, params, request_options=request_options
+        )
 
     def delete(self, plan_id):
         return self._delete("/plans/%s" % plan_id)

diff --git a/shift4/request_options.py b/shift4/request_options.py
@@ -0,0 +1,9 @@
+class RequestOptions:
+    def __init__(self, idempotency_key=None):
+        self.idempotency_key = idempotency_key
+
+    def has_idempotency_key(self):
+        return self.idempotency_key is not None
+
+    def get_idempotency_key(self):
+        return self.idempotency_key
diff --git a/shift4/resource.py b/shift4/resource.py
@@ -13,8 +13,10 @@ def name(self):
     def _get(self, path, params=None, url=None):
         return self.__request("GET", path, params=params, url=url)
 
-    def _post(self, path, json=None, url=None):
-        return self.__request("POST", path, json=json, url=url)
+    def _post(self, path, json=None, url=None, request_options=None):
+        return self.__request(
+            "POST", path, json=json, url=url, request_options=request_options
+        )
 
     def _multipart(self, path, params=None, files=None, url=None):
         return self.__request("POST", path, params=params, files=files, url=url)
@@ -23,14 +25,23 @@ def _delete(self, path, params=None, url=None):
         return self.__request("DELETE", path, params=params, url=url)
 
     @classmethod
-    def __request(cls, method, path, params=None, json=None, files=None, url=None):
+    def __request(
+        cls,
+        method,
+        path,
+        params=None,
+        json=None,
+        files=None,
+        url=None,
+        request_options=None,
+    ):
         if url is None:
             url = api.api_url.rstrip("/")
         resp = requests.request(
             method,
             url=url + path,
             auth=(api.secret_key, ""),
-            headers=cls.__create_headers(),
+            headers=cls.__create_headers(request_options),
             files=files,
             params=params,
             json=json,
@@ -51,12 +62,14 @@ def __request(cls, method, path, params=None, json=None, files=None, url=None):
         )
 
     @classmethod
-    def __create_headers(cls):
+    def __create_headers(cls, request_options=None):
         user_agent = "Shift4-Python/%s (Python/%s.%s.%s)" % (
             __version__,
             sys.version_info.major,
             sys.version_info.minor,
             sys.version_info.micro,
         )
-
-        return {"User-Agent": user_agent}
+        headers = {"User-Agent": user_agent}
+        if request_options is not None and "idempotency_key" in request_options:
+            headers["Idempotency-Key"] = request_options["idempotency_key"]
+        return headers
diff --git a/shift4/subscriptions.py b/shift4/subscriptions.py
@@ -2,14 +2,18 @@
 
 
 class Subscriptions(Resource):
-    def create(self, params):
-        return self._post("/subscriptions", params)
+    def create(self, params, request_options=None):
+        return self._post("/subscriptions", params, request_options=request_options)
 
     def get(self, subscription_id):
         return self._get("/subscriptions/%s" % subscription_id)
 
-    def update(self, subscription_id, params):
-        return self._post("/subscriptions/%s" % subscription_id, params)
+    def update(self, subscription_id, params, request_options=None):
+        return self._post(
+            "/subscriptions/%s" % subscription_id,
+            params,
+            request_options=request_options,
+        )
 
     def cancel(self, subscription_id):
         return self._delete("/subscriptions/%s" % subscription_id)