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

fix(pterodactyl): check if websocket_credentials is None

Browse source
This commit is contained in:
Seaswimmer 2024-03-01 14:18:44 -05:00
parent 684f2559e2
commit b7f5ae644a
Signed by untrusted user: cswimr
GPG key ID: B8953EC01E5C4063
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pterodactyl/websocket.py
View file

@ -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")