Merge pull request #132 from lirenz/master

Channel change merged

Author: danpaquin

gdax/websocket_client.py

@@ -16,11 +16,12 @@
 from pymongo import MongoClient
 
 class WebsocketClient(object):
-    def __init__(self, url="wss://ws-feed.gdax.com", products=None,
-            message_type="subscribe", mongo_collection=None, should_print=True,
-            auth=False, api_key="", api_secret="", api_passphrase=""):
+    def __init__(self, url="wss://ws-feed.gdax.com", products=None, message_type="subscribe", mongo_collection=None, 
+        should_print=True, auth=False, api_key="", api_secret="", api_passphrase="", channels=None):
+
         self.url = url
         self.products = products
+        self.channels = channels
         self.type = message_type
         self.stop = False
         self.error = None
@@ -53,7 +54,12 @@ def _connect(self):
         if self.url[-1] == "/":
             self.url = self.url[:-1]
 
-        sub_params = {'type': 'subscribe', 'product_ids': self.products}
+        if self.channels is None:
+          sub_params = {'type': 'subscribe', 'product_ids': self.products}
+        else:
+          sub_params = {'type': 'subscribe', 'product_ids': self.products, 'channels': self.channels}
+
+
         if self.auth:
             timestamp = str(time.time())
             message = timestamp + 'GET' + '/users/self'