From ebe87decf5bb95cc3a20c0c05afd1ff2a6c16f30 Mon Sep 17 00:00:00 2001 From: AlberLC Date: Fri, 17 May 2024 15:57:24 +0200 Subject: [PATCH] Add telegram restart --- flanabot/bots/flana_bot.py | 3 +++ flanabot/bots/flana_tele_bot.py | 3 ++- flanabot/bots/penalty_bot.py | 3 ++- 3 files changed, 7 insertions(+), 2 deletions(-) diff --git a/flanabot/bots/flana_bot.py b/flanabot/bots/flana_bot.py index 6f0d832..2b03843 100644 --- a/flanabot/bots/flana_bot.py +++ b/flanabot/bots/flana_bot.py @@ -424,6 +424,9 @@ class FlanaBot(Connect4Bot, PenaltyBot, PollBot, ScraperBot, UberEatsBot, Weathe async def _on_ready(self): await super()._on_ready() + if self._is_initialized: + return + flanautils.do_every(multibot_constants.CHECK_OLD_DATABASE_MESSAGES_EVERY_SECONDS, self.check_old_database_actions) for chat in Chat.find({ 'platform': self.platform.value, diff --git a/flanabot/bots/flana_tele_bot.py b/flanabot/bots/flana_tele_bot.py index 7b4a146..6c82a15 100644 --- a/flanabot/bots/flana_tele_bot.py +++ b/flanabot/bots/flana_tele_bot.py @@ -84,7 +84,8 @@ class FlanaTeleBot(TelegramBot, FlanaBot): async def _on_ready(self): await super()._on_ready() - await self._update_whitelist() + if not self._is_initialized: + await self._update_whitelist() # -------------------------------------------------------- # # -------------------- PUBLIC METHODS -------------------- # diff --git a/flanabot/bots/penalty_bot.py b/flanabot/bots/penalty_bot.py index ae17f4b..678b98a 100644 --- a/flanabot/bots/penalty_bot.py +++ b/flanabot/bots/penalty_bot.py @@ -123,7 +123,8 @@ class PenaltyBot(MultiBot, ABC): async def _on_ready(self): await super()._on_ready() - flanautils.do_every(constants.CHECK_PUNISHMENTS_EVERY_SECONDS, self.check_old_punishments) + if not self._is_initialized: + flanautils.do_every(constants.CHECK_PUNISHMENTS_EVERY_SECONDS, self.check_old_punishments) @bot_mentioned @group