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 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
@ -22,6 +21,8 @@ async def establish_websocket_connection(coginstance: Pterodactyl) -> None:
logger.info("Establishing WebSocket connection") logger.info("Establishing WebSocket connection")
websocket_credentials = await retrieve_websocket_credentials(coginstance) 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: 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")