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

Remove UberEatsBot pyperclip

Browse Source
This commit is contained in:
AlberLC
2023-03-30 02:05:32 +02:00
parent aaf17c6877
commit 6be120da17
2 changed files with 5 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
flanabot/bots/ubereats_bot.py
View File

@@ -8,7 +8,6 @@ from collections import defaultdict
import flanautils import flanautils
import playwright.async_api import playwright.async_api
import pyperclip
from multibot import MultiBot, group from multibot import MultiBot, group
import constants import constants
@@ -49,14 +48,7 @@ class UberEatsBot(MultiBot, ABC):
async def _scrape_codes(self, chat: Chat) -> list[str | None]: 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']"): return await page.input_value("input[class='code toCopy']")
return 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'")
return pyperclip.paste()
codes: list[str | None] = [None] * len(chat.ubereats['cookies']) codes: list[str | None] = [None] * len(chat.ubereats['cookies'])
@@ -71,6 +63,7 @@ class UberEatsBot(MultiBot, ABC):
storage_state={'cookies': cookies}, storage_state={'cookies': cookies},
user_agent=flanautils.USER_AGENT user_agent=flanautils.USER_AGENT
) )
context.set_default_timeout(1000)
page = await context.new_page() page = await context.new_page()
await page.goto('https://www.myunidays.com/ES/es-ES/partners/ubereats/access/online') await page.goto('https://www.myunidays.com/ES/es-ES/partners/ubereats/access/online')
@@ -86,11 +79,12 @@ class UberEatsBot(MultiBot, ABC):
else: else:
continue continue
page = context.pages[1] page = context.pages[1]
page.wait_for_load_state()
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 = page.locator("'Obtener nuevo código'")
if new_code_button and await new_code_button.is_enabled(): if 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 (new_code := await get_code()) != code: if (new_code := await get_code()) != code:

1
requirements.txt
View File

@@ -37,7 +37,6 @@ pycryptodomex==3.16.0
pydantic==1.10.4 pydantic==1.10.4
pyee==9.0.4 pyee==9.0.4
pymongo==4.3.3 pymongo==4.3.3
pyperclip==1.8.2
pytube==12.1.2 pytube==12.1.2
pytz==2022.7 pytz==2022.7
requests==2.28.1 requests==2.28.1