chore: добавлена возможность передавать локатор для выбора элемента сигнализирующего о странице антибот защиты

This commit is contained in:
protokey committed 2026-09-18 16:21:07 +04:00
1 parent 6124c604d9
commit ce68d2727c
7 files changed
+36 -11

No files matched your search

+5 -1
View File
@@ -26,6 +26,7 @@ class ChallengeInfo:
provider: str # "cloudflare_turnstile", "datadome", "hcaptcha", "recaptcha_v2", ...
success: bool = False
success_locator: str | None = None # локатор на странице за защитой
detect_locator: str | None = None # локатор на странице за защитой
site_key: str | None = None # для стадии C, чтобы передать в 2captcha
meta: dict | None = None # что-то специфичное для провайдера
@@ -55,6 +56,7 @@ class AntiBotProvider:
_solver_cls: Type[Solver]
_detector_cls: Type[Detector]
_success_locator: str | None = None
_detect_locator: str | None = None
def __init__(self):
self.detector = self._detector_cls()
@@ -64,6 +66,7 @@ class AntiBotProvider:
stage = Stage.INTERSTITIAL,
provider = self.provider,
success_locator = self._success_locator,
detect_locator = self._detect_locator,
)
# обновление info
await self.detector.detect(page, info)
@@ -72,9 +75,10 @@ class AntiBotProvider:
async def solve(self, page: Page, info: ChallengeInfo):
await self.solver.solve(page, info)
def configure(self, success_locator=None, **kwargs):
def configure(self, success_locator=None, detect_locator=None, **kwargs):
self.solver = self._solver_cls(**kwargs)
self._success_locator = success_locator
self._detect_locator = detect_locator
def clear(self):
del self.solver
+1 -1
View File
@@ -42,7 +42,7 @@ class ReCaptchaDetector(Detector):
interactive_marker_selector = "body > div:nth-child(2) > div:nth-child(4) > iframe:nth-child(1)"
# попытка определить что мы вообще на странице блокировки
marker_selector = "#captcha-form"
marker_selector = info.detect_locator or "#captcha-form"
if not await page.query_selector(marker_selector):
log.debug("Page is not a challenge form")
if info.success_locator:
+10 -2
View File
@@ -25,7 +25,11 @@ class ChallengeOutcome:
async def pass_challenges(
antibot_name, page, *, captcha=None, success_locator: str | None = None, timeout_ms=120_000) -> ChallengeOutcome:
antibot_name, page, *,
captcha=None,
success_locator: str | None = None,
detect_locator: str | None = None,
timeout_ms=120_000) -> ChallengeOutcome:
# установка дедлайна
deadline = time.monotonic() + timeout_ms / 1000
@@ -36,7 +40,11 @@ async def pass_challenges(
else:
captcha_solver = None
provider = await select_one(
page, antibot_name, captcha_solver=captcha_solver, success_locator=success_locator)
page, antibot_name,
captcha_solver=captcha_solver,
success_locator=success_locator,
detect_locator=detect_locator,
)
except ProviderNotDetected:
return ChallengeOutcome(False, None, None, "Provider not detected")