From 776b8d009ccbc2f39428cd87e8b4e136f68cdece Mon Sep 17 00:00:00 2001 From: davfsa Date: Thu, 21 Jun 2018 20:19:16 +0200 Subject: [PATCH] Update bot_management.py --- sebimachine/cogs/bot_management.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/sebimachine/cogs/bot_management.py b/sebimachine/cogs/bot_management.py index 2e8d523..7f9de50 100644 --- a/sebimachine/cogs/bot_management.py +++ b/sebimachine/cogs/bot_management.py @@ -10,10 +10,14 @@ class BotManager: if member.bot is False: return else: - # The member is a bot - bot_owner = member.guild.get_member((await self.bot.db_con.fetchval('select owner from bots where id = $1', member.id)) - await bot_owner.add_roles(discord.utils.get(member.guild.roles, name='Bot Developers')) + # Checks if the bot is in the database + if await self.bot.db_con.fetch('select count(*) from bots where id = $1', member.id) != 1: + return await member.kick() + bot_owner = member.guild.get_member((await self.bot.db_con.fetchval('select owner from bots where id = $1', member.id)) + await bot_owner.send("Your bot has been approved and invited") + await bot_owner.add_roles(discord.utils.get(member.guild.roles, name='Bot Developers')) + await member.add_roles(discord.utils.get(member.guild.roles, name='Bots')) try: await member.edit(nick='[' + await self.bot.db_con.fetchval('select prefix from bots where id = $1', member.id)