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

Update UberEatsBot.send_ubereats_code

Browse Source
This commit is contained in:
AlberLC
2023-03-24 00:44:06 +01:00
parent edad87b683
commit 68a3fde4ed
1 changed files with 17 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
flanabot/bots/ubereats_bot.py
View File

@@ -47,7 +47,7 @@ class UberEatsBot(MultiBot, ABC):
if playwright_ := self.task_contexts[chat.id]['playwright']: if playwright_ := self.task_contexts[chat.id]['playwright']:
await playwright_.stop() await playwright_.stop()
async def _scrape_codes(self, chat: Chat) -> list[str]: async def _scrape_codes(self, chat: Chat) -> list[str | None]:
async def get_code() -> str: async def get_code() -> str:
if code_input := await page.query_selector("input[class='code toCopy']"): if code_input := await page.query_selector("input[class='code toCopy']"):
return await code_input.input_value() return await code_input.input_value()
@@ -58,7 +58,7 @@ class UberEatsBot(MultiBot, ABC):
await page.click("'Copiar'") await page.click("'Copiar'")
return pyperclip.paste() return pyperclip.paste()
codes = [] codes: list[str | None] = [None] * len(chat.ubereats['cookies'])
async with playwright.async_api.async_playwright() as playwright_: async with playwright.async_api.async_playwright() as playwright_:
self.task_contexts[chat.id]['playwright'] = playwright_ self.task_contexts[chat.id]['playwright'] = playwright_
@@ -97,7 +97,7 @@ class UberEatsBot(MultiBot, ABC):
code = new_code code = new_code
break break
await asyncio.sleep(0.5) await asyncio.sleep(0.5)
codes.append(code) codes[i] = code
chat.ubereats['cookies'][i] = await context.cookies('https://www.myunidays.com') chat.ubereats['cookies'][i] = await context.cookies('https://www.myunidays.com')
except playwright.async_api.Error: except playwright.async_api.Error:
@@ -139,19 +139,24 @@ class UberEatsBot(MultiBot, ABC):
async def send_ubereats_code(self, chat: Chat, update_next_execution=True): async def send_ubereats_code(self, chat: Chat, update_next_execution=True):
chat.pull_from_database(overwrite_fields=('ubereats',)) chat.pull_from_database(overwrite_fields=('ubereats',))
new_codes = [] codes = await self._scrape_codes(chat)
for code in await self._scrape_codes(chat): for i, code in enumerate(codes):
new_codes.append(code) if code:
if code in chat.ubereats['last_codes']:
if code in chat.ubereats['last_codes']: warning_text = '<i>Código ya enviado anteriormente:</i> '
warning_text = '<i>Código ya enviado anteriormente:</i> ' else:
warning_text = ''
await self.send(f'{warning_text}<code>{code}</code>', chat, silent=True)
else: else:
warning_text = '' try:
await self.send(f'{warning_text}<code>{code}</code>', chat, silent=True) codes[i] = chat.ubereats['last_codes'][i]
chat.ubereats['last_codes'] = new_codes except IndexError:
codes[i] = None
chat.ubereats['last_codes'] = codes
if update_next_execution: if update_next_execution:
chat.ubereats['next_execution'] = datetime.datetime.now(datetime.timezone.utc) + datetime.timedelta(seconds=chat.ubereats['seconds']) 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):