pterodactyl/websocket.py

@@ -1,8 +1,10 @@
 # pylint: disable=cyclic-import
 import json
 import re
+from logging import getLogger
 from typing import Optional, Union
 
+
 import aiohttp
 import discord
 import websockets
@@ -21,7 +23,7 @@ async def establish_websocket_connection(coginstance: Pterodactyl) -> None:
 
     websocket_credentials = await retrieve_websocket_credentials(coginstance)
 
-    async with websockets.connect(websocket_credentials['data']['socket'], origin=base_url, ping_timeout=60) as websocket:
+    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")
 
         auth_message = json.dumps({"event": "auth", "args": [websocket_credentials['data']['token']]})