Skip to content

Commit c1b2664

Browse files
Merge pull request #43 from LewisProjects/development
tiny error fix no real bug
2 parents 14a7362 + ff6c05b commit c1b2664

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

cogs/Levels.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -313,20 +313,24 @@ async def on_level_up(self, msg: Message, level: int):
313313
await self.controller.send_levelup(msg, level)
314314
if await self.is_role_reward(msg.guild, level):
315315
role = await self.get_role_reward(msg.guild, level)
316-
await msg.author.add_roles(role, reason=f"Level up to level {level}")
316+
if role is not None:
317+
await msg.author.add_roles(role, reason=f"Level up to level {level}")
317318

318319
async def is_role_reward(self, guild: Guild, level: int) -> bool:
319320
query = await self.bot.db.execute(
320321
"SELECT EXISTS (SELECT 1 FROM role_rewards WHERE guild_id = ? AND required_lvl = ?)", (guild.id, level)
321322
)
322-
return await query.fetchone() is not None
323+
return bool((await query.fetchone())[0])
323324

324325
async def get_role_reward(self, guild: Guild, level: int) -> Role:
325326

326327
query = await self.bot.db.execute(
327328
"SELECT role_id FROM role_rewards WHERE guild_id = ? AND required_lvl = ?", (guild.id, level)
328329
)
329-
role_id = (await query.fetchone())[0]
330+
try:
331+
role_id = (await query.fetchone())[0]
332+
except TypeError:
333+
return None
330334

331335
return guild.get_role(role_id)
332336

0 commit comments

Comments
 (0)