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

7
flanabot/bots/ubereats_bot.py
View File

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