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

Update flanautils (do_later, do_every)

Browse Source
This commit is contained in:
AlberLC
2023-04-16 09:59:41 +02:00
parent b86fc98377
commit 94fe902d41
4 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
flanabot/bots/flana_bot.py
View File

@@ -230,7 +230,7 @@ class FlanaBot(Connect4Bot, PenaltyBot, PollBot, ScraperBot, UberEatsBot, Weathe
and and
(message.author.is_admin or message.replied_message.author.id == self.id) (message.author.is_admin or message.replied_message.author.id == self.id)
): ):
await flanautils.do_later(flanautils.text_to_time(message.text).total_seconds(), self.delete_message, message.replied_message) flanautils.do_later(flanautils.text_to_time(message.text).total_seconds(), self.delete_message, message.replied_message)
await self.delete_message(message) await self.delete_message(message)
elif message.chat.is_group and self.is_bot_mentioned(message): elif message.chat.is_group and self.is_bot_mentioned(message):
await self.send_negative(message) await self.send_negative(message)
@@ -327,7 +327,7 @@ class FlanaBot(Connect4Bot, PenaltyBot, PollBot, ScraperBot, UberEatsBot, Weathe
async def _on_ready(self): async def _on_ready(self):
await super()._on_ready() await super()._on_ready()
await flanautils.do_every(multibot_constants.CHECK_OLD_DATABASE_MESSAGES_EVERY_SECONDS, self.check_old_database_actions) flanautils.do_every(multibot_constants.CHECK_OLD_DATABASE_MESSAGES_EVERY_SECONDS, self.check_old_database_actions)
for chat in Chat.find({ for chat in Chat.find({
'platform': self.platform.value, 'platform': self.platform.value,
'config.ubereats': {"$exists": True, "$eq": True}, 'config.ubereats': {"$exists": True, "$eq": True},
@@ -340,7 +340,7 @@ class FlanaBot(Connect4Bot, PenaltyBot, PollBot, ScraperBot, UberEatsBot, Weathe
and and
(delta_time := chat.ubereats['next_execution'] - datetime.datetime.now(datetime.timezone.utc)) > datetime.timedelta() (delta_time := chat.ubereats['next_execution'] - datetime.datetime.now(datetime.timezone.utc)) > datetime.timedelta()
): ):
await flanautils.do_later(delta_time, self.start_ubereats, chat) flanautils.do_later(delta_time, self.start_ubereats, chat)
else: else:
await self.start_ubereats(chat) await self.start_ubereats(chat)

2
flanabot/bots/penalty_bot.py
View File

@@ -118,7 +118,7 @@ class PenaltyBot(MultiBot, ABC):
async def _on_ready(self): async def _on_ready(self):
await super()._on_ready() await super()._on_ready()
await flanautils.do_every(constants.CHECK_PUNISHMENTS_EVERY_SECONDS, self.check_old_punishments) flanautils.do_every(constants.CHECK_PUNISHMENTS_EVERY_SECONDS, self.check_old_punishments)
@bot_mentioned @bot_mentioned
@group @group

2
flanabot/bots/scraper_bot.py
View File

@@ -256,7 +256,7 @@ class ScraperBot(MultiBot, ABC):
self.instagram_ban_date = None self.instagram_ban_date = None
bot_message = await self.send('Ban de Instagram reseteado.', message) bot_message = await self.send('Ban de Instagram reseteado.', message)
await self.delete_message(message) await self.delete_message(message)
await flanautils.do_later(multibot_constants.COMMAND_MESSAGE_DURATION, self.delete_message, bot_message) flanautils.do_later(multibot_constants.COMMAND_MESSAGE_DURATION, self.delete_message, bot_message)
async def _on_scraping( async def _on_scraping(
self, self,

2
flanabot/bots/ubereats_bot.py
View File

@@ -170,7 +170,7 @@ class UberEatsBot(MultiBot, ABC):
await self._cancel_scraping_task(chat) await self._cancel_scraping_task(chat)
chat.config['ubereats'] = True chat.config['ubereats'] = True
chat.save(pull_overwrite_fields=('ubereats',)) chat.save(pull_overwrite_fields=('ubereats',))
self.task_contexts[chat.id]['task'] = await flanautils.do_every(chat.ubereats['seconds'], self.send_ubereats_code, chat, do_first_now=send_code_now) self.task_contexts[chat.id]['task'] = flanautils.do_every(chat.ubereats['seconds'], self.send_ubereats_code, chat, do_first_now=send_code_now)
async def stop_ubereats(self, chat: Chat): async def stop_ubereats(self, chat: Chat):
await self._cancel_scraping_task(chat) await self._cancel_scraping_task(chat)