pterodactyl/websocket.py

@@ -4,7 +4,6 @@ import re
 from logging import getLogger
 from typing import Optional, Union
 
-
 import aiohttp
 import discord
 import websockets
@@ -22,6 +21,8 @@ async def establish_websocket_connection(coginstance: Pterodactyl) -> None:
     logger.info("Establishing WebSocket connection")
 
     websocket_credentials = await retrieve_websocket_credentials(coginstance)
+    if not websocket_credentials:
+        return logger.error("Failed to retrieve WebSocket credentials. WebSocket connection not established")
 
     async with websockets.connect(websocket_credentials['data']['socket'], origin=base_url, ping_timeout=60, logger=getLogger("red.sea.pterodactyl.websocket")) as websocket:
         logger.info("WebSocket connection established")