75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
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()
|