1
0
Fork 0

Add support of disabled extension for rorec

This commit is contained in:
Ethanell 2020-11-05 14:59:06 +01:00
parent b24aebc906
commit 297a05d9f4

View file

@ -7,8 +7,9 @@ from discord import Embed, RawReactionActionEvent, RawBulkMessageDeleteEvent, Ra
from discord.ext.commands import BadArgument from discord.ext.commands import BadArgument
from administrator import db from administrator import db
from administrator.check import is_enabled
from administrator.logger import logger from administrator.logger import logger
from administrator.utils import event_is_enabled
extension_name = "rorec" extension_name = "rorec"
logger = logger.getChild(extension_name) logger = logger.getChild(extension_name)
@ -40,6 +41,7 @@ class RoRec(commands.Cog):
await (await ctx.channel.fetch_message(ctx.message.id)).remove_reaction(emoji, self.bot.user) await (await ctx.channel.fetch_message(ctx.message.id)).remove_reaction(emoji, self.bot.user)
@commands.group("rorec", pass_context=True) @commands.group("rorec", pass_context=True)
@is_enabled()
@commands.guild_only() @commands.guild_only()
@commands.has_permissions(manage_roles=True) @commands.has_permissions(manage_roles=True)
async def rorec(self, ctx: commands.Context): async def rorec(self, ctx: commands.Context):
@ -204,6 +206,8 @@ class RoRec(commands.Cog):
@commands.Cog.listener() @commands.Cog.listener()
async def on_raw_reaction_add(self, payload: RawReactionActionEvent): async def on_raw_reaction_add(self, payload: RawReactionActionEvent):
s = db.Session() s = db.Session()
if payload.guild_id and not event_is_enabled(self.qualified_name, payload.guild_id, s):
return
m = s.query(db.RoRec).filter(db.RoRec.message == payload.message_id).first() m = s.query(db.RoRec).filter(db.RoRec.message == payload.message_id).first()
s.close() s.close()
if m and payload.member.id != self.bot.user.id: if m and payload.member.id != self.bot.user.id: