forked from cswimr/SeaCogs
fix(pterodactyl): fixed circular import
This commit is contained in:
parent
2bf8629e4c
commit
7f8d70ccae
2 changed files with 2 additions and 2 deletions
|
@ -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
|
||||
|
|
|
@ -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()
|
||||
|
|
Loading…
Reference in a new issue