Add Streaming Bug fix.

Author: areed1192

td/stream.py

@@ -530,7 +530,7 @@ def stream(self, print_to_console: bool = True) -> None:
         self.print_to_console = print_to_console
 
         # Connect to the Websocket.
-        self.loop.run_until_complete(self._connect(pipeline_start=False))
+        self.loop.run_until_complete(self._connect())
 
         # Send the Request.
         asyncio.ensure_future(self._send_message(self._build_data_request()))
@@ -597,7 +597,7 @@ async def close_stream(self) -> None:
         #     except asyncio.CancelledError:
         #         print("main(): cancel_me is cancelled now")
 
-    async def _connect(self, pipeline_start: bool = True) -> websockets.WebSocketClientProtocol:
+    async def _connect(self) -> websockets.WebSocketClientProtocol:
         """Connects the Client to the TD Websocket.
 
         Connecting to webSocket server websockets.client.connect 
@@ -621,21 +621,13 @@ async def _connect(self, pipeline_start: bool = True) -> websockets.WebSocketCli
         # Create a connection.
         self.connection = await websockets.client.connect(self.websocket_url)
 
-        # check it before sending it back.
-        if await self._check_connection() and pipeline_start == True:
+        # See if we are connected.
+        is_connected = await self._check_connection()
 
-            # Login to the stream.
+        # If we are connected then login.
+        if is_connected:
             await self._send_message(login_request)
-            await self._receive_message(return_value=True)
             return self.connection
-        
-        else:
-
-            # Login to the stream.
-            await self._send_message(login_request)
-            await self._receive_message(return_value=False)
-            return self.connection
-
 
     async def _check_connection(self) -> bool:
         """Determines if we have an active connection
@@ -717,7 +709,7 @@ async def _receive_message(self, return_value: bool = False) -> dict:
                     print('')
                     print(message_decoded)
                     print('-'*20)
-                    print('')                  
+                    print('')         
 
             except websockets.exceptions.ConnectionClosed: