Merge pull request #42 from Devination/patch-1

Fixed JSON str vs bytes error:
alpacahq · Dec 5, 2018 · be87d4c · be87d4c
2 parents 1e376a0 + c9ca8cb
commit be87d4c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/alpaca_trade_api/stream2.py b/alpaca_trade_api/stream2.py
@@ -27,6 +27,8 @@ async def _connect(self):
             }
         }))
         r = await ws.recv()
+        if isinstance(r, bytes):
+            r = r.decode('utf-8')
         msg = json.loads(r)
         # TODO: check unauthorized
         self._ws = ws
@@ -40,6 +42,8 @@ async def _consume_msg(self):
         try:
             while True:
                 r = await ws.recv()
+                if isinstance(r, bytes):
+                    r = r.decode('utf-8')
                 msg = json.loads(r)
                 stream = msg.get('stream')
                 if stream is not None: