From cda045f123bcddab5b601a5f98a44d563460138b Mon Sep 17 00:00:00 2001
From: jaredgoldman <jaredgoldmandev@gmail.com>
Date: Wed, 8 Mar 2023 23:45:51 -0500
Subject: [PATCH] Abstract header gen to seperate function

---
 piker/brokers/kucoin.py | 68 +++++++++++++++++++++++------------------
 1 file changed, 38 insertions(+), 30 deletions(-)

diff --git a/piker/brokers/kucoin.py b/piker/brokers/kucoin.py
index e1e9ebd9..1a7861cb 100644
--- a/piker/brokers/kucoin.py
+++ b/piker/brokers/kucoin.py
@@ -77,47 +77,55 @@ class Client:
             self._key_secret = config["key_secret"]
             self._key_passphrase = config["key_passphrase"]
 
+    def _gen_auth_req_headers(
+        self,
+        action: Literal["POST", "GET", "PUT", "DELETE"],
+        endpoint: str,
+        api_v: str = "v2",
+    ):
+        now = int(time.time() * 1000)
+        path = f'/api/{api_v}{endpoint}'
+        str_to_sign = str(now) + action + path
+
+        # Add headers to request if authenticated
+        signature = base64.b64encode(
+            hmac.new(
+                self._key_secret.encode('utf-8'),
+                str_to_sign.encode('utf-8'),
+                hashlib.sha256
+            ).digest()
+        )
+
+        passphrase = base64.b64encode(
+            hmac.new(
+                self._key_secret.encode('utf-8'),
+                self._key_passphrase.encode('utf-8'),
+                hashlib.sha256
+            ).digest()
+        )
+
+        return {
+            "KC-API-SIGN": signature,
+            "KC-API-TIMESTAMP": str(now),
+            "KC-API-KEY": self._key_id,
+            "KC-API-PASSPHRASE": passphrase,
+            "KC-API-KEY-VERSION": "2"
+        }
+
     async def _request(
         self,
         action: Literal["POST", "GET", "PUT", "DELETE"],
         endpoint: str,
         api_v: str = "v2",
     ) -> Any:
-
-        now = int(time.time() * 1000)
-        path = f'/api/{api_v}{endpoint}'
-        str_to_sign = str(now) + action + path
         headers = {}
 
-        # Add headers to request if authenticated
         if self._authenticated:
-            signature = base64.b64encode(
-                hmac.new(
-                    self._key_secret.encode('utf-8'),
-                    str_to_sign.encode('utf-8'),
-                    hashlib.sha256
-                ).digest()
-            )
+            headers = self._gen_auth_req_headers(action, endpoint, api_v)
 
-            passphrase = base64.b64encode(
-                hmac.new(
-                    self._key_secret.encode('utf-8'),
-                    self._key_passphrase.encode('utf-8'),
-                    hashlib.sha256
-                ).digest()
-            )
-
-            headers = {
-                "KC-API-SIGN": signature,
-                "KC-API-TIMESTAMP": str(now),
-                "KC-API-KEY": self._key_id,
-                "KC-API-PASSPHRASE": passphrase,
-                "KC-API-KEY-VERSION": "2"
-            }
-
-        api_url = f"https://api.kucoin.com{path}"
+        api_url = f"https://api.kucoin.com/api/{api_v}{endpoint}"
         res = await asks.request(action, api_url, headers=headers)
-        # breakpoint()
+
         if "data" in res.json():
             return res.json()["data"]
         else: