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

Add ubereats_next_execution

Browse Source
This commit is contained in:
AlberLC
2023-03-06 07:34:09 +01:00
parent 7fee5f382c
commit 61daaa387a
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
flanabot/bots/ubereats_bot.py
View File

@@ -118,7 +118,7 @@ class UberEatsBot(MultiBot, ABC):
time = flanautils.text_to_time(message.text)
if not time:
bot_state_message = await self.send(random.choice(constants.SCRAPING_PHRASES), message)
await self.send_ubereats_code(message.chat)
await self.send_ubereats_code(message.chat, update_next_execution=False)
await self.delete_message(bot_state_message)
return
@@ -136,7 +136,7 @@ class UberEatsBot(MultiBot, ABC):
# -------------------------------------------------------- #
# -------------------- PUBLIC METHODS -------------------- #
# -------------------------------------------------------- #
async def send_ubereats_code(self, chat: Chat):
async def send_ubereats_code(self, chat: Chat, update_next_execution=True):
new_codes = []
for code in await self._scrape_codes(chat):
new_codes.append(code)
@@ -148,6 +148,7 @@ class UberEatsBot(MultiBot, ABC):
await self.send(f'{warning_text} <code>{code}</code>', chat, silent=True)
chat.ubereats_last_codes = new_codes
if update_next_execution:
chat.ubereats_next_execution = datetime.datetime.now(datetime.timezone.utc) + datetime.timedelta(seconds=chat.ubereats_seconds)
chat.save()