pterodactyl/ptero.py

@@ -2,7 +2,6 @@ import asyncio
 import aiohttp
 import discord
 import requests
-from discord import ui
 from redbot.core import commands, app_commands, Config
 
 class Pterodactyl(commands.Cog):
@@ -109,14 +108,14 @@ class Pterodactyl(commands.Cog):
             passed_info = [old_startup_args[0], old_startup_args[1], headers, put_data, new_put_data, interaction_message]
             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))
 
-    class UpdateButtons(ui.View):
+    class UpdateButtons(discord.ui.View):
         def __init__(self, timeout, passed_info):
             super().__init__()
             self.passed_info = passed_info
 
-        @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(interaction.guild, "power"), headers=self.passed_info[2], json={"signal": "stop"})
+        @discord.ui.button(label="Yes", style=discord.ButtonStyle.success)
+        async def yes_button(self, button:discord.ui.Button, interaction:discord.Interaction):
+            requests.post(await Pterodactyl.get_url(interaction.guild, "power"), headers=headers, json={"signal": "stop"})
             await interaction.response.edit_message("Server stopping...", view=None)
             while True:
                 async with aiohttp.ClientSession() as session:
@@ -159,8 +158,8 @@ class Pterodactyl(commands.Cog):
                             await asyncio.sleep(1)
                             continue
 
-        @ui.button(label="No", style=discord.ButtonStyle.danger)
-        async def no_button(self, button:ui.Button, interaction:discord.Interaction):
+        @discord.ui.button(label="No", style=discord.ButtonStyle.danger)
+        async def no_button(self, button:discord.ui.Button, interaction:discord.Interaction):
             for button in self.children:
                 button.disabled = True
             await interaction.response.edit_message(content=f"Command cancelled.", view=self)
@@ -267,3 +266,14 @@ class Pterodactyl(commands.Cog):
             else:
                 send = f"Configuration changed:\n{message}"
             await interaction.response.send_message(send, ephemeral=True)
+
+
+import requests
+headers = {
+    "Authorization": "Bearer ptlc_9BVTvT41F7mNtG7YIqaxhOySf4Z8YDmVYwaITwwIbA7",
+    "Content-Type": "application/json",
+    "Accept": "Application/vnd.pterodactyl.v1+json"
+}
+request = requests.get("https://mc.bloom.host/api/client/servers/d6d05b41-4da5-4c5b-9ee6-6e25f61a596b/startup", headers=headers)
+request_dict = request.json()
+print(request_dict)