Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
2 changed files
with
176 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,152 @@ | ||
| from __future__ import absolute_import | ||
|
|
||
| import requests | ||
|
|
||
| from ..exceptions import ( | ||
| CaptchaServiceUnavailable, | ||
| CaptchaAPIError, | ||
| CaptchaTimeout, | ||
| CaptchaParameter, | ||
| CaptchaBadJobID | ||
| ) | ||
|
|
||
|
|
||
| try: | ||
| import polling2 | ||
| except ImportError: | ||
| raise ImportError("Please install the python module 'polling2' via pip") | ||
|
|
||
| from . import Captcha | ||
|
|
||
|
|
||
| class captchaSolver(Captcha): | ||
| def __init__(self): | ||
| self.host = 'https://api.captchaai.io' | ||
| self.session = requests.Session() | ||
| super(captchaSolver, self).__init__('captchaai') | ||
|
|
||
| # ------------------------------------------------------------------------------- # | ||
|
|
||
| @staticmethod | ||
| def checkErrorStatus(response, request_type): | ||
| if response.status_code in [500, 502]: | ||
| raise CaptchaServiceUnavailable(f'CaptchaAI: Server Side Error {response.status_code}') | ||
|
|
||
| try: | ||
| rPayload = response.json() | ||
| except Exception: | ||
| return | ||
|
|
||
| if rPayload.get('errorDescription', False) and 'Current system busy' not in rayload['errorDescription']: | ||
| raise CaptchaAPIError( | ||
| f"CaptchaAI: {request_type} -> {rPayload.get('errorDescription')}" | ||
| ) | ||
|
|
||
| # ------------------------------------------------------------------------------- # | ||
|
|
||
| def requestJob(self, jobID): | ||
| if not jobID: | ||
| raise CaptchaBadJobID("CaptchaAI: Error bad job id to request task result.") | ||
|
|
||
| def _checkRequest(response): | ||
| self.checkErrorStatus(response, 'getTaskResult') | ||
| try: | ||
| rPayload = response.json() | ||
| if response.ok: | ||
| if rPayload.get("solution", {}).get('gRecaptchaResponse'): | ||
| return True | ||
| except Exception: | ||
| pass | ||
| return None | ||
|
|
||
| response = polling2.poll( | ||
| lambda: self.session.post( | ||
| f'{self.host}/getTaskResult', | ||
| json={ | ||
| 'clientKey': self.api_key, | ||
| 'taskId': jobID | ||
| }, | ||
| timeout=30 | ||
| ), | ||
| check_success=_checkRequest, | ||
| step=5, | ||
| timeout=180 | ||
| ) | ||
|
|
||
| if response: | ||
| try: | ||
| rPayload = response.json() | ||
| if rPayload.get('solution', {}).get('gRecaptchaResponse'): | ||
| return rPayload['solution']['gRecaptchaResponse'] | ||
| except Exception: | ||
| pass | ||
|
|
||
| raise CaptchaTimeout( | ||
| "CaptchaAI: Error failed to solve Captcha." | ||
| ) | ||
|
|
||
| # ------------------------------------------------------------------------------- # | ||
|
|
||
| def requestSolve(self, captchaType, url, siteKey): | ||
| def _checkRequest(response): | ||
| self.checkErrorStatus(response, 'createTask') | ||
| try: | ||
| rPayload = response.json() | ||
| if response.ok: | ||
| if rPayload.get("taskId", False): | ||
| return True | ||
| except Exception: | ||
| pass | ||
| return None | ||
|
|
||
| response = polling2.poll( | ||
| lambda: self.session.post( | ||
| f'{self.host}/createTask', | ||
| json={ | ||
| 'clientKey': self.api_key, | ||
| 'appId': '9E717405-8C70-49B3-B277-7C2F2196484B', | ||
| 'task': { | ||
| 'type': 'HCaptchaTaskProxyless', | ||
| 'websiteURL': url, | ||
| 'websiteKey': siteKey | ||
| } | ||
| }, | ||
| allow_redirects=False, | ||
| timeout=30 | ||
| ), | ||
| check_success=_checkRequest, | ||
| step=5, | ||
| timeout=180 | ||
| ) | ||
|
|
||
| if response: | ||
| rPayload = response.json() | ||
| if rPayload.get('taskId'): | ||
| return rPayload['taskId'] | ||
|
|
||
| raise CaptchaBadJobID( | ||
| 'CaptchaAI: Error no job id was returned.' | ||
| ) | ||
|
|
||
| # ------------------------------------------------------------------------------- # | ||
|
|
||
| def getCaptchaAnswer(self, captchaType, url, siteKey, captchaParams): | ||
| if not captchaParams.get('api_key'): | ||
| raise CaptchaParameter("CaptchaAI: Missing api_key parameter.") | ||
|
|
||
| self.api_key = captchaParams.get('api_key') | ||
|
|
||
| try: | ||
| jobID = self.requestSolve(captchaType, url, siteKey) | ||
| return self.requestJob(jobID) | ||
| except polling2.TimeoutException: | ||
| raise CaptchaTimeout( | ||
| f"captchaAI: Captcha solve (task ID: {jobID}) took to long." | ||
| ) | ||
|
|
||
| raise CaptchaAPIError('CaptchaAI: Job Failure.') | ||
|
|
||
|
|
||
| # ------------------------------------------------------------------------------- # | ||
|
|
||
| captchaSolver() |