chore: update websocket implementation to new asyncio api

freqtrade/rpc/api_server/ws/proxy.py

@@ -1,7 +1,7 @@
 from typing import Any
 
 from fastapi import WebSocket as FastAPIWebSocket
-from websockets.legacy.client import WebSocketClientProtocol as WebSocket
+from websockets.asyncio.client import ClientConnection as WebSocket
 
 from freqtrade.rpc.api_server.ws.types import WebSocketType
 

freqtrade/rpc/api_server/ws/types.py

@@ -1,7 +1,7 @@
 from typing import Any, TypeVar
 
 from fastapi import WebSocket as FastAPIWebSocket
-from websockets.legacy.client import WebSocketClientProtocol as WebSocket
+from websockets.asyncio.client import ClientConnection as WebSocket
 
 
 WebSocketType = TypeVar("WebSocketType", FastAPIWebSocket, WebSocket)

freqtrade/rpc/external_message_consumer.py

@@ -193,7 +193,7 @@ async def _create_connection(self, producer: Producer, lock: asyncio.Lock):
                 ws_url = f"{scheme}://{host}:{port}/api/v1/message/ws?token={token}"
 
                 # This will raise InvalidURI if the url is bad
-                async with websockets.legacy.client.connect(
+                async with websockets.connect(
                     ws_url, max_size=self.message_size_limit, ping_interval=None
                 ) as ws:
                     async with create_channel(ws, channel_id=name, send_throttle=0.5) as channel:

scripts/ws_client.py

@@ -253,8 +253,7 @@ async def create_client(
         except (
             socket.gaierror,
             ConnectionRefusedError,
-            websockets.exceptions.InvalidStatusCode,
-            websockets.exceptions.InvalidMessage,
+            websockets.exceptions.InvalidHandshake,
         ) as e:
             logger.error(f"Connection Refused - {e} retrying in {sleep_time}s")
             await asyncio.sleep(sleep_time)