Add check to rolegiver to verify that roles property exists

This commit is contained in:
Zeva Rose 2022-05-04 23:53:49 -06:00
parent 7ccc7ea6d2
commit ab71fa8b3c

View file

@ -49,13 +49,15 @@ class RolegiverCog(Cog):
return return
setting = await Rolegiver.find_one(q(guild=ctx.guild.id)) setting = await Rolegiver.find_one(q(guild=ctx.guild.id))
if setting and role.id in setting.roles: if setting and setting.roles and role.id in setting.roles:
await ctx.send("Role already in rolegiver", ephemeral=True) await ctx.send("Role already in rolegiver", ephemeral=True)
return return
if not setting: if not setting:
setting = Rolegiver(guild=ctx.guild.id, roles=[]) setting = Rolegiver(guild=ctx.guild.id, roles=[])
setting.roles = setting.roles or []
if len(setting.roles) >= 20: if len(setting.roles) >= 20:
await ctx.send("You can only have 20 roles in the rolegiver", ephemeral=True) await ctx.send("You can only have 20 roles in the rolegiver", ephemeral=True)
return return