From 2a7bf32aab2a175fb960e10a5d41ea39b7802983 Mon Sep 17 00:00:00 2001 From: Zevaryx Date: Mon, 26 Jul 2021 19:26:07 -0600 Subject: [PATCH] Add on_message_edit event listener --- jarvis/events/message.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/jarvis/events/message.py b/jarvis/events/message.py index df89fff..28a62cd 100644 --- a/jarvis/events/message.py +++ b/jarvis/events/message.py @@ -18,6 +18,7 @@ class MessageEventHandler(object): def __init__(self, bot): self.bot = bot self.bot.add_listener(self.on_message) + self.bot.add_listener(self.on_message_edit) async def autopurge(self, message: Message): autopurge = Autopurge.get( @@ -222,11 +223,8 @@ class MessageEventHandler(object): await self.checks(message) await self.bot.process_commands(message) - async def on_message_edit(self, message: Message): - if ( - not isinstance(message.channel, DMChannel) - and not message.author.bot - ): - await self.massmention(message) - await self.roleping(message) - await self.checks(message) + async def on_message_edit(self, before: Message, after: Message): + if not isinstance(after.channel, DMChannel) and not after.author.bot: + await self.massmention(after) + await self.roleping(after) + await self.checks(after)