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)