Добавлен модуль CustomCaptchaTask

Добавлена типизация переменных в captcha_handler методе FunCaptcha

python3_anticaptcha/CustomCaptchaTask.py

@@ -0,0 +1,144 @@
+import requests
+import time
+import aiohttp
+import asyncio
+
+from .config import create_task_url, app_key, user_agent_data
+from .get_answer import get_sync_result, get_async_result
+
+
+class CustomCaptchaTask:
+	def __init__(self, anticaptcha_key: str, sleep_time: int = 5, assignment: str = None, forms: dict = None, callbackUrl: str = None):
+		"""
+		Модуль отвечает за решение CustomCaptchaTask
+		Параметр userAgent рандомно берётся из актульного списка браузеров-параметров
+		:param anticaptcha_key: Ключ от АнтиКапчи
+		:param sleep_time: Время ожидания решения
+        :param assignment: Опишите что работник должен сделать на английском языке
+        :param forms: Произвольная форма в формате JSON
+        :param callbackUrl: URL для решения капчи с ответом через callback
+		"""
+		if sleep_time < 5:
+			raise ValueError(f'Параметр `sleep_time` должен быть не менее 5. Вы передали - {sleep_time}')
+		self.sleep_time = sleep_time
+		
+		# Пайлоад для создания задачи
+		self.task_payload = {"clientKey": anticaptcha_key,
+							 "task":
+								 {
+									 "type": "CustomCaptchaTask",
+								 },
+							 "softId": app_key
+                             }
+        # заполняем поле с заданием пользователю        
+        if assignment:
+            self.task_payload['task'].update({'assignment': assignment})
+        # заполняем поле с произвольной формой в формате JSON
+        if forms:
+            self.task_payload['task'].update({'forms': forms})
+		
+        # задаём callbackUrl если передан
+		if callbackUrl:
+			self.task_payload.update({'callbackUrl': callbackUrl})
+
+		# пайлоад для получения ответа сервиса
+		self.result_payload = {"clientKey": anticaptcha_key}
+		
+		
+	# Работа с капчёй
+	def captcha_handler(self, imageUrl: str):
+		"""
+		Метод получает ссылку изображение для задания
+		:param imageUrl: URL картинки
+		:return: Возвращает ответ сервера в виде JSON(ответ так же можно глянуть в документации антикапчи)
+		"""
+		self.task_payload['task'].update({"imageUrl": imageUrl})
+		# Отправляем на антикапча параметры фанкапич,
+		# в результате получаем JSON ответ содержащий номер решаемой капчи
+		captcha_id = requests.post(create_task_url, json=self.task_payload).json()
+
+		# Проверка статуса создания задачи, если создано без ошибок - извлекаем ID задачи, иначе возвращаем ответ сервера
+		if captcha_id['errorId'] == 0:
+			captcha_id = captcha_id["taskId"]
+			# обновляем пайлоад на получение решения капчи
+			self.result_payload.update({"taskId": captcha_id})
+		else:
+			return captcha_id
+        # если передан параметр `callbackUrl` - не ждём решения капчи а возвращаем незаполненный ответ
+		if self.task_payload.get('callbackUrl'):
+			return self.result_payload
+        
+		else:
+            # Ждем решения капчи
+			time.sleep(self.sleep_time)
+			return get_sync_result(result_payload = self.result_payload, sleep_time = self.sleep_time)
+
+
+
+class aioCustomCaptchaTask:
+	def __init__(self, anticaptcha_key: str, imageUrl: str, sleep_time: int = 5, assignment: str = None, forms: dict = None, callbackUrl: str = None):
+		"""
+		Модуль отвечает за решение CustomCaptchaTask
+		Параметр userAgent рандомно берётся из актульного списка браузеров-параметров
+		:param anticaptcha_key: Ключ от АнтиКапчи
+		:param sleep_time: Время ожидания решения
+        :param assignment: Опишите что работник должен сделать на английском языке
+        :param forms: Произвольная форма в формате JSON
+        :param callbackUrl: URL для решения капчи с ответом через callback
+		"""
+		if sleep_time < 5:
+			raise ValueError(f'Параметр `sleep_time` должен быть не менее 5. Вы передали - {sleep_time}')
+		self.sleep_time = sleep_time
+		
+		# Пайлоад для создания задачи
+		self.task_payload = {"clientKey": anticaptcha_key,
+							 "task":
+								 {
+									 "type": "CustomCaptchaTask",
+								 },
+							 "softId": app_key
+                             }
+        # заполняем поле с заданием пользователю        
+        if assignment:
+            self.task_payload['task'].update({'assignment': assignment})
+        # заполняем поле с произвольной формой в формате JSON
+        if forms:
+            self.task_payload['task'].update({'forms': forms})
+		
+        # задаём callbackUrl если передан
+		if callbackUrl:
+			self.task_payload.update({'callbackUrl': callbackUrl})
+
+		# пайлоад для получения ответа сервиса
+		self.result_payload = {"clientKey": anticaptcha_key}
+
+	# Работа с капчёй
+	async def captcha_handler(self, imageUrl: str):
+		"""
+		Метод получает ссылку изображение для задания
+		:param imageUrl: URL картинки
+		:return: Возвращает ответ сервера в виде JSON(ответ так же можно глянуть в документации антикапчи)
+		"""
+		self.task_payload['task'].update({{"imageUrl": imageUrl}})
+		# Отправляем на антикапча параметры фанкапич,
+		# в результате получаем JSON ответ содержащий номер решаемой капчи
+		async with aiohttp.ClientSession() as session:
+			async with session.post(create_task_url, json=self.task_payload) as resp:
+				captcha_id = await resp.json()
+		
+		# Проверка статуса создания задачи, если создано без ошибок - извлекаем ID задачи, иначе возвращаем ответ сервера
+		if captcha_id['errorId'] == 0:
+			captcha_id = captcha_id["taskId"]
+			# обновляем пайлоад на получение решения капчи
+			self.result_payload.update({"taskId": captcha_id})
+		else:
+			return captcha_id
+
+        # если передан параметр `callbackUrl` - не ждём решения капчи а возвращаем незаполненный ответ
+		if self.task_payload.get('callbackUrl'):
+			return self.result_payload
+            
+		else:
+            # Ждем решения капчи
+			await asyncio.sleep(self.sleep_time)
+			return await get_async_result(result_payload = self.result_payload, sleep_time = self.sleep_time)

python3_anticaptcha/FunCaptchaTask.py

@@ -51,7 +51,7 @@ class FunCaptchaTask:
 				self.task_payload['task'].update({key: kwargs[key]})
 		
 	# Работа с капчёй
-	def captcha_handler(self, websiteURL, websitePublicKey, **kwargs):
+	def captcha_handler(self, websiteURL: str, websitePublicKey: str, **kwargs):
 		"""
 		Метод получает ссылку на страницу на которпой расположена капча и ключ капчи
 		:param websiteURL: Ссылка на страницу с капчёй
@@ -125,7 +125,7 @@ class aioFunCaptchaTask:
 				self.task_payload['task'].update({key: kwargs[key]})
 	
 	# Работа с капчёй
-	async def captcha_handler(self, websiteURL, websitePublicKey):
+	async def captcha_handler(self, websiteURL: str, websitePublicKey: str):
 		"""
 		Метод получает ссылку на страницу на которпой расположена капча и ключ капчи
 		:param websiteURL: Ссылка на страницу с капчёй