cswimr/SeaCogs
cswimr/SeaCogs
1
1
Fork 1
Code Issues 10 Pull requests 4 Projects 1 Packages 1 Wiki Activity Actions

fix(pterodactyl): use a custom logger for the websocket instead of websockets.client
Some checks failed
Actions / Lint Code (Ruff & Pylint) (pull_request) Failing after 21s
Details
Actions / Build Documentation (MkDocs) (pull_request) Successful in 25s
Details

Browse source
This commit is contained in:
Seaswimmer 2024-03-01 14:04:10 -05:00
parent 837bf4aa34
commit 8b1e42716b
Signed by: 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']]})