spawnrys/flanabot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add telegram restart

Browse Source
This commit is contained in:
AlberLC
2024-05-17 16:06:09 +02:00
parent ebe87decf5
commit 564a65c5a1
3 changed files with 22 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
flanabot/bots/flana_bot.py
View File

@@ -423,26 +423,25 @@ class FlanaBot(Connect4Bot, PenaltyBot, PollBot, ScraperBot, UberEatsBot, Weathe
await super()._on_new_message_raw(message, whitelist_callbacks, blacklist_callbacks)
async def _on_ready(self):
await super()._on_ready()
if self._is_initialized:
return
if not self._is_initialized:
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,
'config.ubereats': {"$exists": True, "$eq": True},
'ubereats.cookies': {"$exists": True, "$ne": []}
}):
chat = await self.get_chat(chat.id)
chat.pull_from_database(overwrite_fields=('_id', 'config', 'ubereats'))
if (
chat.ubereats['next_execution']
and
(delta_time := chat.ubereats['next_execution'] - datetime.datetime.now(datetime.timezone.utc)) > datetime.timedelta()
):
flanautils.do_later(delta_time, self.start_ubereats, chat)
else:
await self.start_ubereats(chat)
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,
'config.ubereats': {"$exists": True, "$eq": True},
'ubereats.cookies': {"$exists": True, "$ne": []}
}):
chat = await self.get_chat(chat.id)
chat.pull_from_database(overwrite_fields=('_id', 'config', 'ubereats'))
if (
chat.ubereats['next_execution']
and
(delta_time := chat.ubereats['next_execution'] - datetime.datetime.now(datetime.timezone.utc)) > datetime.timedelta()
):
flanautils.do_later(delta_time, self.start_ubereats, chat)
else:
await self.start_ubereats(chat)
await super()._on_ready()
@inline(False)
async def _on_recover_message(self, message: Message):

3
flanabot/bots/flana_tele_bot.py
View File

@@ -83,10 +83,11 @@ class FlanaTeleBot(TelegramBot, FlanaBot):
await super()._on_new_message_raw(message, whitelist_callbacks, blacklist_callbacks)
async def _on_ready(self):
await super()._on_ready()
if not self._is_initialized:
await self._update_whitelist()
await super()._on_ready()
# -------------------------------------------------------- #
# -------------------- PUBLIC METHODS -------------------- #
# -------------------------------------------------------- #

3
flanabot/bots/penalty_bot.py
View File

@@ -122,10 +122,11 @@ class PenaltyBot(MultiBot, ABC):
await self.punish(user, message, flanautils.text_to_time(await self.filter_mention_ids(message.text, message)), message)
async def _on_ready(self):
await super()._on_ready()
if not self._is_initialized:
flanautils.do_every(constants.CHECK_PUNISHMENTS_EVERY_SECONDS, self.check_old_punishments)
await super()._on_ready()
@bot_mentioned
@group
@admin(send_negative=True)