GalacticFactory/GalacticCogs
GalacticFactory/GalacticCogs
Archived
1
1
Fork 0
Code Issues Pull requests Activity

Refactor Pterodactyl cog to use aiohttp #2

Merged
cswimr merged 26 commits from aiohttp-refactor into main 2023-07-24 09:29:55 -04:00
Conversation 0 Commits 26 Files changed 1 +2 -5
1 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit 7727f789f2 - Show all commits

7
pterodactyl/ptero.py
View file

@ -20,7 +20,7 @@ class Pterodactyl(commands.Cog):
startup_arguments=None startup_arguments=None
) )
async def get_session(self, guild, endpoint = None): async def get_session(self, guild):
if await self.config.guild(guild).server_id() is None: if await self.config.guild(guild).server_id() is None:
raise LookupError("Server ID not set.") raise LookupError("Server ID not set.")
elif await self.config.guild(guild).base_url() is None: elif await self.config.guild(guild).base_url() is None:
@ -32,10 +32,7 @@ class Pterodactyl(commands.Cog):
"Content-Type": "application/json", "Content-Type": "application/json",
"Accept": "application/json" "Accept": "application/json"
} }
url = f"https://{await self.config.guild(guild).base_url()}/api/client/servers/{await self.config.guild(guild).server_id()}" async with aiohttp.ClientSession(headers=headers) as session:
if endpoint:
url += '/' + endpoint
async with aiohttp.ClientSession(url=url, headers=headers) as session:
return session return session
async def get_url(self, guild, endpoint = None): async def get_url(self, guild, endpoint = None):