client implementation, minor fixes

scripts/test_ws_client.py

@@ -1,74 +0,0 @@
-import asyncio
-import logging
-import socket
-from typing import Any
-
-import websockets
-
-from freqtrade.enums import RPCMessageType
-from freqtrade.rpc.api_server.ws.channel import WebSocketChannel
-
-
-logging.basicConfig(level=logging.DEBUG)
-logger = logging.getLogger(__name__)
-
-
-def compose_consumer_request(type_: str, data: Any):
-    return {"type": type_, "data": data}
-
-
-async def _client():
-    # Trying to recreate multiple topic issue. Wait until first whitelist message,
-    # then CTRL-C to get the status message.
-    topics = [RPCMessageType.WHITELIST, RPCMessageType.STATUS]
-    try:
-        while True:
-            try:
-                url = "ws://localhost:8080/api/v1/message/ws?token=testtoken"
-                async with websockets.connect(url) as ws:
-                    channel = WebSocketChannel(ws)
-
-                    logger.info("Connection successful")
-                    # Tell the producer we only want these topics
-                    await channel.send(compose_consumer_request("subscribe", topics))
-
-                    while True:
-                        try:
-                            data = await asyncio.wait_for(
-                                channel.recv(),
-                                timeout=5
-                            )
-                            logger.info(f"Data received - {data}")
-                        except (asyncio.TimeoutError, websockets.exceptions.ConnectionClosed):
-                            # We haven't received data yet. Check the connection and continue.
-                            try:
-                                # ping
-                                ping = await channel.ping()
-                                await asyncio.wait_for(ping, timeout=2)
-                                logger.debug(f"Connection to {url} still alive...")
-                                continue
-                            except Exception:
-                                logger.info(
-                                    f"Ping error {url} - retrying in 5s")
-                                await asyncio.sleep(2)
-                                break
-
-            except (socket.gaierror, ConnectionRefusedError):
-                logger.info("Connection Refused - retrying connection in 5s")
-                await asyncio.sleep(2)
-                continue
-            except websockets.exceptions.InvalidStatusCode as e:
-                logger.error(f"Connection Refused - {e}")
-                await asyncio.sleep(2)
-                continue
-
-    except (asyncio.CancelledError, KeyboardInterrupt):
-        pass
-
-
-def main():
-    asyncio.run(_client())
-
-
-if __name__ == "__main__":
-    main()