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

Fix UberEats task stderr

Browse Source
This commit is contained in:
AlberLC
2023-03-04 05:12:33 +01:00
parent 154f02a1b6
commit 81854375d1
1 changed files with 30 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
flanabot/bots/ubereats_bot.py
View File

@@ -32,40 +32,41 @@ class UberEatsBot(MultiBot, ABC):
@staticmethod
async def _scrape_code(chat: Chat) -> str:
async with playwright.async_api.async_playwright() as playwright_:
async with await playwright_.chromium.launch() as browser:
context: playwright.async_api.BrowserContext = await browser.new_context()
await context.add_cookies(chat.ubereats_cookies)
page = await context.new_page()
await page.goto('https://www.myunidays.com/ES/es-ES/partners/ubereats/access/online')
with flanautils.suppress_low_level_stderr():
async with playwright.async_api.async_playwright() as playwright_:
async with await playwright_.chromium.launch() as browser:
context: playwright.async_api.BrowserContext = await browser.new_context()
await context.add_cookies(chat.ubereats_cookies)
page = await context.new_page()
await page.goto('https://www.myunidays.com/ES/es-ES/partners/ubereats/access/online')
if button := await page.query_selector("button[class='button highlight']"):
await button.click()
else:
await page.click("'Revelar código'")
while len(context.pages) != 2:
await asyncio.sleep(0.5)
page = context.pages[1]
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'")
if new_code_button and await new_code_button.is_enabled():
await new_code_button.click()
await page.wait_for_load_state('networkidle')
if code_input := await page.query_selector("input[class='code toCopy']"):
code = await code_input.input_value()
else:
if button := await page.query_selector("button[class='copy button quarternary']"):
if button := await page.query_selector("button[class='button highlight']"):
await button.click()
else:
await page.click("'Copiar'")
code = pyperclip.paste()
await page.click("'Revelar código'")
while len(context.pages) != 2:
await asyncio.sleep(0.5)
page = context.pages[1]
chat.ubereats_cookies = await context.cookies('https://www.myunidays.com')
chat.save()
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'")
if new_code_button and await new_code_button.is_enabled():
await new_code_button.click()
await page.wait_for_load_state('networkidle')
return code
if code_input := await page.query_selector("input[class='code toCopy']"):
code = await code_input.input_value()
else:
if button := await page.query_selector("button[class='copy button quarternary']"):
await button.click()
else:
await page.click("'Copiar'")
code = pyperclip.paste()
chat.ubereats_cookies = await context.cookies('https://www.myunidays.com')
chat.save()
return code
# ---------------------------------------------- #
# HANDLERS #