update ws client

This commit is contained in:
Timothy Pogue 2022-11-15 22:44:02 -07:00
parent 86e094e39b
commit 1380ddd066

View File

@ -199,6 +199,7 @@ async def create_client(
host, host,
port, port,
token, token,
scheme='ws',
name='default', name='default',
protocol=ClientProtocol(), protocol=ClientProtocol(),
sleep_time=10, sleep_time=10,
@ -211,13 +212,14 @@ async def create_client(
:param host: The host :param host: The host
:param port: The port :param port: The port
:param token: The websocket auth token :param token: The websocket auth token
:param scheme: `ws` for most connections, `wss` for ssl
:param name: The name of the producer :param name: The name of the producer
:param **kwargs: Any extra kwargs passed to websockets.connect :param **kwargs: Any extra kwargs passed to websockets.connect
""" """
while 1: while 1:
try: try:
websocket_url = f"ws://{host}:{port}/api/v1/message/ws?token={token}" websocket_url = f"{scheme}://{host}:{port}/api/v1/message/ws?token={token}"
logger.info(f"Attempting to connect to {name} @ {host}:{port}") logger.info(f"Attempting to connect to {name} @ {host}:{port}")
async with websockets.connect(websocket_url, **kwargs) as ws: async with websockets.connect(websocket_url, **kwargs) as ws:
@ -304,6 +306,7 @@ async def _main(args):
producer['host'], producer['host'],
producer['port'], producer['port'],
producer['ws_token'], producer['ws_token'],
'wss' if producer['secure'] else 'ws',
producer['name'], producer['name'],
sleep_time=sleep_time, sleep_time=sleep_time,
ping_timeout=ping_timeout, ping_timeout=ping_timeout,