From 4d60281a23b99af093feee1f7cbdfe6820538b88 Mon Sep 17 00:00:00 2001 From: AlberLC Date: Mon, 16 Jan 2023 03:31:41 +0100 Subject: [PATCH] Add delay to FlanaBot._on_delete --- flanabot/bots/flana_bot.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/flanabot/bots/flana_bot.py b/flanabot/bots/flana_bot.py index 911213b..cb52f35 100644 --- a/flanabot/bots/flana_bot.py +++ b/flanabot/bots/flana_bot.py @@ -1,5 +1,6 @@ __all__ = ['FlanaBot'] +import asyncio import datetime import random from abc import ABC @@ -170,10 +171,15 @@ class FlanaBot(Connect4Bot, PenaltyBot, PollBot, ScraperBot, WeatherBot, MultiBo @inline(False) async def _on_delete(self, message: Message): if message.replied_message: - if message.replied_message.author.id == self.id: - await self.delete_message(message.replied_message) + if ( + self.is_bot_mentioned(message) + and + (message.author.is_admin or message.replied_message.author.id == self.id) + ): if message.chat.is_group: await self.delete_message(message) + await asyncio.sleep(flanautils.text_to_time(message.text).total_seconds()) + await self.delete_message(message.replied_message) elif message.chat.is_group and self.is_bot_mentioned(message): await self.send_negative(message) elif (