ISSUE-128: refactored to prevent duplicate code

Author: kzielinski

gdax/authenticated_client.py

@@ -12,6 +12,7 @@
 import json
 from requests.auth import AuthBase
 from gdax.public_client import PublicClient
+from gdax.gdax_auth import GdaxAuth
 
 
 class AuthenticatedClient(PublicClient):
@@ -288,28 +289,4 @@ def get_report(self, report_id=""):
     def get_trailing_volume(self):
         r = requests.get(self.url + "/users/self/trailing-volume", auth=self.auth, timeout=30)
         # r.raise_for_status()
-        return r.json()
-
-
-class GdaxAuth(AuthBase):
-    # Provided by gdax: https://docs.gdax.com/#signing-a-message
-    def __init__(self, api_key, secret_key, passphrase):
-        self.api_key = api_key
-        self.secret_key = secret_key
-        self.passphrase = passphrase
-
-    def __call__(self, request):
-        timestamp = str(time.time())
-        message = timestamp + request.method + request.path_url + (request.body or '')
-        message = message.encode('ascii')
-        hmac_key = base64.b64decode(self.secret_key)
-        signature = hmac.new(hmac_key, message, hashlib.sha256)
-        signature_b64 = base64.b64encode(signature.digest())
-        request.headers.update({
-            'Content-Type': 'Application/JSON',
-            'CB-ACCESS-SIGN': signature_b64,
-            'CB-ACCESS-TIMESTAMP': timestamp,
-            'CB-ACCESS-KEY': self.api_key,
-            'CB-ACCESS-PASSPHRASE': self.passphrase
-        })
-        return request
+        return r.json()

gdax/gdax_auth.py

@@ -0,0 +1,34 @@
+import hmac
+import hashlib
+import time
+import base64
+from requests.auth import AuthBase
+
+
+class GdaxAuth(AuthBase):
+    # Provided by gdax: https://docs.gdax.com/#signing-a-message
+    def __init__(self, api_key, secret_key, passphrase):
+        self.api_key = api_key
+        self.secret_key = secret_key
+        self.passphrase = passphrase
+
+    def __call__(self, request):
+        timestamp = str(time.time())
+        message = timestamp + request.method + request.path_url + (request.body or '')
+        request.headers.update(get_auth_headers(timestamp, message, self.api_key, self.secret_key,
+                                                self.passphrase))
+        return request
+
+
+def get_auth_headers(timestamp, message, api_key, secret_key, passphrase):
+    message = message.encode('ascii')
+    hmac_key = base64.b64decode(secret_key)
+    signature = hmac.new(hmac_key, message, hashlib.sha256)
+    signature_b64 = base64.b64encode(signature.digest()).decode('utf-8')
+    return {
+        'Content-Type': 'Application/JSON',
+        'CB-ACCESS-SIGN': signature_b64,
+        'CB-ACCESS-TIMESTAMP': timestamp,
+        'CB-ACCESS-KEY': api_key,
+        'CB-ACCESS-PASSPHRASE': passphrase
+    }

gdax/websocket_client.py

@@ -14,6 +14,7 @@
 from threading import Thread
 from websocket import create_connection, WebSocketConnectionClosedException
 from pymongo import MongoClient
+from gdax.gdax_auth import get_auth_headers
 
 
 class WebsocketClient(object):
@@ -62,14 +63,7 @@ def _connect(self):
         if self.auth:
             timestamp = str(time.time())
             message = timestamp + 'GET' + '/users/self'
-            message = message.encode('ascii')
-            hmac_key = base64.b64decode(self.api_secret)
-            signature = hmac.new(hmac_key, message, hashlib.sha256)
-            signature_b64 = base64.b64encode(signature.digest()).decode("utf-8")
-            sub_params['CB-ACCESS-SIGN'] = signature_b64
-            sub_params['CB-ACCESS-KEY'] = self.api_key
-            sub_params['CB-ACCESS-PASSPHRASE'] = self.api_passphrase
-            sub_params['CB-ACCESS-TIMESTAMP'] = timestamp
+            sub_params.update(get_auth_headers(timestamp, message, self.api_key,  self.api_secret, self.api_passphrase))
 
         self.ws = create_connection(self.url)
         self.ws.send(json.dumps(sub_params))