Improve UberEatsBot new code obtaining

This commit is contained in:
AlberLC
2023-03-12 05:21:47 +01:00
parent 4ebcade424
commit 44dcb3d987

View File

@@ -87,17 +87,16 @@ class UberEatsBot(MultiBot, ABC):
continue continue
page = context.pages[1] page = context.pages[1]
last_code = await get_code() code = await get_code()
if not (new_code_button := await page.query_selector("button[class='getNewCode button secondary']")): if not (new_code_button := await page.query_selector("button[class='getNewCode button secondary']")):
new_code_button = await page.query_selector("'Obtener nuevo código'") new_code_button = await page.query_selector("'Obtener nuevo código'")
if new_code_button and await new_code_button.is_enabled(): if new_code_button and await new_code_button.is_enabled():
await new_code_button.click() await new_code_button.click()
for _ in range(5): for _ in range(5):
if (code := await get_code()) != last_code: if (new_code := await get_code()) != code:
code = new_code
break break
await asyncio.sleep(0.5) await asyncio.sleep(0.5)
else:
code = last_code
codes.append(code) codes.append(code)
chat.ubereats['cookies'][i] = await context.cookies('https://www.myunidays.com') chat.ubereats['cookies'][i] = await context.cookies('https://www.myunidays.com')