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
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']]})