pterodactyl/pterodactyl.py

@@ -2,7 +2,6 @@ import asyncio
 import json
 from typing import Optional
 
-
 import discord
 import websockets
 from pydactyl import PterodactylClient
@@ -12,7 +11,6 @@ from redbot.core.utils.chat_formatting import box
 
 from pterodactyl.config import config
 from pterodactyl.logger import logger
-from pterodactyl.websocket import establish_websocket_connection
 
 
 class Pterodactyl(commands.Cog):
@@ -32,6 +30,7 @@ class Pterodactyl(commands.Cog):
         await self.client._session.close() # pylint: disable=protected-access
 
     def get_task(self) -> asyncio.Task:
+        from pterodactyl.websocket import establish_websocket_connection
         task = self.bot.loop.create_task(establish_websocket_connection(self), name="Pterodactyl Websocket Connection")
         task.add_done_callback(self.error_callback)
         return task

pterodactyl/websocket.py

@@ -13,6 +13,7 @@ from pterodactyl.logger import logger
 from pterodactyl.pterodactyl import Pterodactyl
 
 
+
 async def establish_websocket_connection(coginstance: Pterodactyl) -> None:
     logger.debug("Establishing WebSocket connection")
     base_url = await config.base_url()