Merge pull request #272 from alpacahq/a_bit_more_verbose

A bit more verbose ws connection

Author: Shlomi Kushchi

alpaca_trade_api/polygon/streamconn.py

@@ -43,6 +43,7 @@ async def connect(self):
                     .format(msg))
             )
         await self._dispatch(msg)
+        logging.info(f"connected to: {self._endpoint}")
         if await self.authenticate():
             self._consume_task = asyncio.ensure_future(self._consume_msg())
         else:

alpaca_trade_api/stream2.py

@@ -2,6 +2,8 @@
 import json
 import os
 import re
+from asyncio import CancelledError
+
 import websockets
 from .common import get_base_url, get_data_url, get_credentials, URL
 from .entity import Account, Entity, trade_mapping, agg_mapping, quote_mapping
@@ -57,7 +59,7 @@ async def _connect(self):
 
         self._ws = ws
         await self._dispatch('authorized', msg)
-
+        logging.info(f"connected to: {self._endpoint}")
         self._consume_task = asyncio.ensure_future(self._consume_msg())
 
     async def consume(self):
@@ -289,7 +291,8 @@ def run(self, initial_channels: List[str] = []):
                 logging.info("Exiting on Interrupt")
                 should_renew = False
             except Exception as e:
-                logging.error(f"error while consuming ws messages: {e}")
+                m = 'consume cancelled' if isinstance(e, CancelledError) else e
+                logging.error(f"error while consuming ws messages: {m}")
                 loop.run_until_complete(self.close(should_renew))
                 if loop.is_running():
                     loop.close()