Rosie-Val/SeaCogs
1
0
Fork 0
Code Pull requests Activity

fix(pterodactyl): use a custom logger for the websocket instead of websockets.client

Browse source
This commit is contained in:
Seaswimmer 2024-03-01 14:04:10 -05:00
parent 837bf4aa34
commit 8b1e42716b
Signed by untrusted user: cswimr
GPG key ID: B8953EC01E5C4063
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pterodactyl/websocket.py
View file

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