🐛 fix owner check

URN · Apr 28, 2023 · cb92fe1 · cb92fe1
1 parent f325313
commit cb92fe1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/wopr/__main__.py b/wopr/__main__.py
@@ -35,6 +35,7 @@ class BotBase(commands.Bot):
     """A base class for the bot to allow for custom attributes."""
 
     start_time: datetime
+    owners: list[int]
 
 
 logging.basicConfig(handlers=[InterceptHandler()], level=logging.INFO, force=True)
@@ -57,6 +58,9 @@ async def on_ready() -> None:
     bot.tree.copy_global_to(guild=MY_GUILD)
     await bot.tree.sync(guild=MY_GUILD)
 
+    for owner in await bot.application_info().team.members:
+        bot.owners.append(owner.id)
+
 
 @bot.tree.command()
 async def ping(interaction: discord.Interaction) -> None:

diff --git a/wopr/exts/extensions.py b/wopr/exts/extensions.py
@@ -16,7 +16,7 @@ def __init__(self, bot: commands.Bot):
 
     async def interaction_check(self, interaction: discord.Interaction) -> bool:
         """Check if the user is an administrator."""
-        return interaction.user.id in self.bot.owner_ids
+        return interaction.user.id in self.bot.owners
 
     @app_commands.command()
     async def load(self, interaction: discord.Interaction, extension: str) -> None: