diff --git a/src/antibot/base.py b/src/antibot/base.py index 7cb6cdd..71ded06 100644 --- a/src/antibot/base.py +++ b/src/antibot/base.py @@ -97,7 +97,7 @@ class AntiBotProvider: async def detect(self, page: Page) -> ChallengeInfo: info = ChallengeInfo( - stage=Stage.INTERSTITIAL, + stage=None, # noqa provider=self.provider, success_locator=self._success_locator, detect_locator=self._detect_locator, @@ -107,6 +107,7 @@ class AntiBotProvider: return info async def solve(self, page: Page, info: ChallengeInfo): + assert info.stage, "Unknown stage" await self.solver.solve(page, info) def configure(self, success_locator=None, detect_locator=None, **kwargs): diff --git a/src/antibot/known_providers/recaptcha.py b/src/antibot/known_providers/recaptcha.py index b3949eb..ece9423 100644 --- a/src/antibot/known_providers/recaptcha.py +++ b/src/antibot/known_providers/recaptcha.py @@ -114,6 +114,7 @@ class ReCaptchaDetector(Detector): iframe = page.frame_locator("iframe[title='reCAPTCHA']") # ожидаем когда будет загружен чекбокс + info.stage = Stage.INTERSTITIAL checkbox = iframe.locator(checkbox_selector) try: await checkbox.wait_for(state="attached")