diff --git a/reginaldCog/reginald.py b/reginaldCog/reginald.py index 2b14ba2..e64683c 100644 --- a/reginaldCog/reginald.py +++ b/reginaldCog/reginald.py @@ -34,14 +34,10 @@ async def predicate(ctx): def has_admin_role(): async def predicate(ctx): - #janitor_role_id = 672156832323600396 - #has_janitor_role = any(role.id == janitor_role_id for role in ctx.author.roles) # Uncomment this line has_admin_permission = ctx.author.guild_permissions.administrator - #return has_janitor_role or has_admin_permission return has_admin_permission return commands.check(predicate) - @commands.guild_only() @commands.has_permissions(manage_guild=True) @commands.command(help="Set the OpenAI API key") diff --git a/trafficCog/__init__.py b/trafficCog/__init__.py index 55dbf36..6426d01 100644 --- a/trafficCog/__init__.py +++ b/trafficCog/__init__.py @@ -2,4 +2,5 @@ from redbot.core.bot import Red async def setup(bot: Red): - await bot.add_cog(TrafficCog(bot)) \ No newline at end of file + cog = TrafficCog(bot) + await bot.add_cog(cog) \ No newline at end of file