def ws(sdk_client_fs: ADCMClient, max_conn=10):
while max_conn:
try:
ws_conn = websocket.create_connection(
url="ws://" + prep_url(sdk_client_fs.url) + "/ws/event/",
subprotocols=["adcm", sdk_client_fs.api_token()],
timeout=15)
except websocket.WebSocketBadStatusException:
pass
else:
return ws_conn
max_conn -= 1
break
def websocket_connection(sdk_client_fs: ADCMClient, max_conn=10):
"""Create WS connection"""
last_error = None
while max_conn:
try:
ws_conn = websocket.create_connection(
url="ws://" + prep_url(sdk_client_fs.url) + "/ws/event/",
subprotocols=["adcm", sdk_client_fs.api_token()],
timeout=15,
)
except websocket.WebSocketBadStatusException as error:
last_error = error
else:
return ws_conn
max_conn -= 1
raise ValueError(
f"Could not create websocket connection in {max_conn} attempts. "
f"Last error is:\n{last_error}")
