From ddc4690ee54af6ebd925ed58f7d72bd677282da0 Mon Sep 17 00:00:00 2001 From: Elnath Date: Sat, 26 Dec 2020 22:32:09 +0100 Subject: [PATCH] Check admin: now also accepts bot owner --- VocalMaisBot.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/VocalMaisBot.py b/VocalMaisBot.py index 6a0a84b..eb0d1bc 100755 --- a/VocalMaisBot.py +++ b/VocalMaisBot.py @@ -25,10 +25,10 @@ class VocalMaisBot(discord.Client): except json.JSONDecodeError as e: logger.critical(f"Impossible to parse watched channels file, JSON error: {e}") sys.exit(1) + self.owner = None # Fetched by on_ready - @staticmethod - async def _check_administrator_for_command(message: discord.Message) -> bool: - if message.author.guild_permissions.administrator: + async def _check_administrator_for_command(self, message: discord.Message) -> bool: + if message.author.guild_permissions.administrator or message.author == self.owner: return True else: await message.channel.send(f":dragon_face: Only an administrator can control me!") @@ -49,6 +49,8 @@ class VocalMaisBot(discord.Client): logger.info("Connected and ready!") logger.debug(f"Logged as {self.user}") + self.owner = (await self.application_info()).owner + oauth_url = discord.utils.oauth_url( self.user.id, discord.Permissions(manage_channels = True, read_messages = True, send_messages = True, move_members = True)