diff --git a/pterodactyl/ptero.py b/pterodactyl/ptero.py index 04d49b3..54c884a 100644 --- a/pterodactyl/ptero.py +++ b/pterodactyl/ptero.py @@ -3,6 +3,7 @@ import aiohttp import discord import requests from discord import ui +from discord.ext import commands from redbot.core import commands, app_commands, Config class Pterodactyl(commands.Cog): @@ -109,7 +110,8 @@ class Pterodactyl(commands.Cog): "headers": headers, "updater_startup_vars": updater_startup_vars, "old_startup_vars": old_startup_vars, - "interaction": interaction + "interaction": interaction, + "interaction_context": await commands.Context.from_interaction(interaction), } await interaction_message.edit(content="The server is already running! Are you sure you'd like to stop the server for updates?", view=self.UpdateButtons(timeout=180, passed_info=passed_info)) @@ -120,7 +122,7 @@ class Pterodactyl(commands.Cog): @ui.button(label="Yes", style=discord.ButtonStyle.success) async def yes_button(self, button:ui.Button, interaction:discord.Interaction): - requests.post(await Pterodactyl.get_url(self, interaction.guild, "power"), headers=self.passed_info['headers'], json={"signal": "stop"}) + requests.post(await Pterodactyl.get_url(self, self.passed_info['interaction_context'].guild, "power"), headers=self.passed_info['headers'], json={"signal": "stop"}) await self.passed_info['interaction'].edit_original_response("Server stopping...", view=None) while True: async with aiohttp.ClientSession() as session: