flifloo/Administrator-py
Archived
1
0
Fork 0
Code Activity

Add support of disabled extension for purge

Browse source
This commit is contained in:
Ethanell 2020-11-05 14:54:00 +01:00
parent 2ba73ea5d0
commit 234e8e1877
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
extensions/purge.py
View file

@ -3,8 +3,9 @@ from asyncio import sleep
from discord.ext import commands from discord.ext import commands
from discord import Embed, RawReactionActionEvent from discord import Embed, RawReactionActionEvent
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 = "purge" extension_name = "purge"
logger = logger.getChild(extension_name) logger = logger.getChild(extension_name)
@ -19,6 +20,7 @@ class Purge(commands.Cog):
return "Purge all messages between the command and the next add reaction" return "Purge all messages between the command and the next add reaction"
@commands.group("purge", pass_context=True) @commands.group("purge", pass_context=True)
@is_enabled()
@commands.guild_only() @commands.guild_only()
@commands.has_permissions(manage_messages=True) @commands.has_permissions(manage_messages=True)
async def purge(self, ctx: commands.Context): async def purge(self, ctx: commands.Context):
@ -45,6 +47,8 @@ class Purge(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):
if payload.guild_id: if payload.guild_id:
if not event_is_enabled(self.qualified_name, payload.guild_id):
return
user = self.bot.get_user(payload.user_id) user = self.bot.get_user(payload.user_id)
message = await self.bot.get_guild(payload.guild_id).get_channel(payload.channel_id)\ message = await self.bot.get_guild(payload.guild_id).get_channel(payload.channel_id)\
.fetch_message(payload.message_id) .fetch_message(payload.message_id)