From 7e80d6e30471005fe4ebb7591d5be708aff116b2 Mon Sep 17 00:00:00 2001 From: unknown Date: Sat, 3 Jun 2023 17:23:31 +0200 Subject: [PATCH] Reverted reginaldCog --- reginaldCog/reginald.py | 19 ++++++------------- 1 file changed, 6 insertions(+), 13 deletions(-) diff --git a/reginaldCog/reginald.py b/reginaldCog/reginald.py index 0a0ef2d..509a26b 100644 --- a/reginaldCog/reginald.py +++ b/reginaldCog/reginald.py @@ -9,7 +9,6 @@ import aiohttp from io import BytesIO from PIL import Image import tempfile -import configparser from openai import OpenAIError from redbot.core import Config, commands @@ -17,11 +16,7 @@ from redbot.core import Config, commands class ReginaldCog(commands.Cog): def __init__(self, bot): self.bot = bot - config = configparser.ConfigParser() - config.read('config.ini') - - identifier_id = int(config['DEFAULT']['identifier_id']) - self.config = Config.get_conf(self, identifier=identifier_id) + self.config = Config.get_conf(self, identifier=71717171171717) default_global = { "openai_model": "gpt-3.5-turbo" } @@ -31,20 +26,18 @@ class ReginaldCog(commands.Cog): self.config.register_global(**default_global) self.config.register_guild(**default_guild) - self.allowed_role_id = int(config['DEFAULT']['allowed_role_id']) - def has_allowed_role(): - def predicate(ctx): - return any(role.id == ctx.cog.allowed_role_id for role in ctx.author.roles) + async def predicate(ctx): + kanium_role_id = 280260875678515200 + return any(role.id == kanium_role_id for role in ctx.author.roles) return commands.check(predicate) def has_admin_role(): - def predicate(ctx): + async def predicate(ctx): has_admin_permission = ctx.author.guild_permissions.administrator 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") @@ -184,4 +177,4 @@ class ReginaldCog(commands.Cog): def setup(bot): cog = ReginaldCog(bot) - bot.add_cog(cog) + bot.add_cog(cog) \ No newline at end of file