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

Add Pterodactyl cog #19

Merged
cswimr merged 139 commits from pterodactyl into main 2024-03-02 00:07:42 -05:00
Conversation 0 Commits 139 Files changed 22 +2 -2
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit 239f7cf613 - Show all commits

4
pterodactyl/pterodactyl.py
View file

@ -62,10 +62,10 @@ class Pterodactyl(commands.Cog):
self.logger.debug("Authentication message sent") self.logger.debug("Authentication message sent")
self.websocket = websocket self.websocket = websocket
current_status = ''
while True: while True:
message = await websocket.recv() message = await websocket.recv()
current_status = ''
if json.loads(message)['event'] in ['token expiring', 'token expired']: if json.loads(message)['event'] in ['token expiring', 'token expired']:
self.logger.debug("Received token expiring/expired event. Refreshing token.") self.logger.debug("Received token expiring/expired event. Refreshing token.")
websocket_credentials = client.servers.get_websocket(server_id) websocket_credentials = client.servers.get_websocket(server_id)
@ -77,7 +77,7 @@ class Pterodactyl(commands.Cog):
self.logger.debug("Authentication successful") self.logger.debug("Authentication successful")
if json.loads(message)['event'] == 'console output' and await self.config.console_channel() is not None: if json.loads(message)['event'] == 'console output' and await self.config.console_channel() is not None:
if current_status != 'starting' and current_status != 'stopping' and current_status != 'restarting': if current_status == 'started' or current_status == 'offline' or current_status == '':
channel = self.bot.get_channel(await self.config.console_channel()) channel = self.bot.get_channel(await self.config.console_channel())
if channel is not None: if channel is not None:
content = self.remove_ansi_escape_codes(json.loads(message)['args'][0][:1900]) content = self.remove_ansi_escape_codes(json.loads(message)['args'][0][:1900])