Merge pull request danpaquin#4 from adamgilman/master

danpaquin · web-flow · commit f9004796ed99 · 2016-11-05T19:33:29.000-04:00
Support multiple products via websockets stream
diff --git a/GDAX/WebsocketClient.py b/GDAX/WebsocketClient.py
@@ -20,7 +20,11 @@ def __init__(self, ws_url="wss://ws-feed.gdax.com", product_id="BTC-USD"):
     def setup(self):
         self.open()
         self.ws = create_connection(self.url)
-        subParams = json.dumps({"type": "subscribe", "product_id": self.product_id})
+        if type(self.product_id) is list:
+            #product_ids - plural for multiple products
+            subParams = json.dumps({"type": "subscribe", "product_ids": self.product_id})
+        else:
+            subParams = json.dumps({"type": "subscribe", "product_id": self.product_id})
         self.ws.send(subParams)
         self.listen()
 
diff --git a/README.md b/README.md
@@ -193,6 +193,8 @@ authClient.withdraw(withdrawParams)
 
 ### WebsocketClient
 If you would like to receive real-time market updates, you must subscribe to the [websocket feed](https://docs.gdax.com/#websocket-feed).
+
+#### Subscribe to a single product
 ```python
 import GDAX
 # Paramters are optional
@@ -201,6 +203,15 @@ wsClient = GDAX.WebsocketClient(ws_url="wss://ws-feed.gdax.com", product_id="BTC
 wsClient.close()
 ```
 
+#### Subscribe to multiple products
+```python
+import GDAX
+# Paramters are optional
+wsClient = GDAX.WebsocketClient(ws_url="wss://ws-feed.gdax.com", product_id=["BTC-USD", "ETH-USD"])
+# Do other stuff...
+wsClient.close()
+```
+
 ### WebsocketClient Methods
 The ```WebsocketClient``` subscribes in a separate thread upon initialization.  There are three methods which you could overwrite (before initialization) so it can react to the data streaming in.  The current client is a template used for illustration purposes only.
 
@@ -228,4 +239,4 @@ The following projects are suggested to improve the functionality of this projec
 - [**danielktaylor/PyLimitBook**](https://github.com/danielktaylor/PyLimitBook)
    *Python implementation of fast limit-order book.*
 - [**PierreRochard/coinbase-exchange-order-book**](https://github.com/PierreRochard/coinbase-exchange-order-book)
-    *Real-time Coinbase Exchange order book + basic market maker bot (Python3)*
+    *Real-time Coinbase Exchange order book + basic market maker bot (Python3)*
