From fd3bef9240f02b5711eebc5dc6e4f0188049cf31 Mon Sep 17 00:00:00 2001 From: Le Codex <40335983+LeCodex@users.noreply.github.com> Date: Sun, 5 Apr 2020 17:26:51 +0200 Subject: [PATCH] Added emjoi functionnality to remove error messages early --- modules/errors/__init__.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/modules/errors/__init__.py b/modules/errors/__init__.py index dc257a4..3cbd24a 100644 --- a/modules/errors/__init__.py +++ b/modules/errors/__init__.py @@ -85,11 +85,19 @@ class MainClass(BaseClassPython): self.objects.save_object('errorsList', self.errorsList) # Wait 60 seconds and delete message - await asyncio.sleep(60) + #await asyncio.sleep(60) try: - channel = self.client.get_channel(msg_id["channel_id"]) - delete_message = await channel.fetch_message(msg_id["msg_id"]) - await delete_message.delete() + # channel = self.client.get_channel(msg_id["channel_id"]) + # delete_message = await channel.fetch_message(msg_id["msg_id"]) + # await delete_message.delete() + await message.add_reaction("🗑️") + + try: + reaction, user = await self.client.wait_for('reaction_add', timeout=60.0, check=lambda r, u: r.emoji == "🗑️" and not u.bot) + except asyncio.TimeoutError: + await message.delete() + else: + await reaction.message.delete() except: raise finally: