Kanium-PUBLIC/KaniumCogs
5
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

Please work :(

Browse Source
This commit is contained in:
unknown 2023-06-03 18:35:15 +02:00
parent 3d7cf3e2f9
commit f6b94847d9
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
reginaldCog/reginald.py
View File

@ -12,6 +12,11 @@ import tempfile
from openai import OpenAIError from openai import OpenAIError
from redbot.core import Config, commands from redbot.core import Config, commands
def role_check():
async def predicate(ctx):
cog = ctx.bot.get_cog("ReginaldCog")
return await cog.has_admin_role(ctx) or await cog.has_allowed_role(ctx)
return commands.check(predicate)
class ReginaldCog(commands.Cog): class ReginaldCog(commands.Cog):
def __init__(self, bot): def __init__(self, bot):
@ -69,7 +74,7 @@ class ReginaldCog(commands.Cog):
await ctx.send("OpenAI API key set successfully.") await ctx.send("OpenAI API key set successfully.")
@commands.guild_only() @commands.guild_only()
@commands.check(role_check()) @role_check() # Use role_check without parentheses
@commands.command(help="Ask Reginald a question") @commands.command(help="Ask Reginald a question")
@commands.cooldown(1, 10, commands.BucketType.user) # 10 second cooldown per user @commands.cooldown(1, 10, commands.BucketType.user) # 10 second cooldown per user
async def reginald(self, ctx, *, prompt=None): async def reginald(self, ctx, *, prompt=None):