461 lines
24 KiB
Python
461 lines
24 KiB
Python
import time
|
||
import asyncio
|
||
import hashlib
|
||
import os
|
||
import base64
|
||
|
||
import requests
|
||
import aiohttp
|
||
|
||
from python3_anticaptcha import (
|
||
create_task_url,
|
||
app_key,
|
||
get_sync_result,
|
||
get_async_result,
|
||
ParamError,
|
||
ReadError,
|
||
)
|
||
|
||
|
||
class SquareNetTextTask:
|
||
"""
|
||
Данный метод подходит для решения капчи-изображение.
|
||
Подробней информацию смотрите в методе 'captcha_handler' и '__init__'
|
||
"""
|
||
|
||
def __init__(
|
||
self,
|
||
anticaptcha_key: str,
|
||
sleep_time: int = 5,
|
||
save_format: str = "temp",
|
||
callbackUrl: str = None,
|
||
):
|
||
"""
|
||
Инициализация нужных переменных, создание папки для изображений и кэша
|
||
После завершения работы - удаляются временные фалйы и папки
|
||
:param anticaptcha_key: АПИ ключ капчи из кабинета пользователя
|
||
:param sleep_time: Вермя ожидания решения капчи
|
||
:param save_format: Формат в котором будет сохраняться изображение, либо как временный файл - 'temp',
|
||
либо как обычное изображение в папку созданную библиотекой - 'const'.
|
||
:param callbackUrl: URL для решения капчи с ответом через callback
|
||
"""
|
||
if sleep_time < 5:
|
||
raise ValueError(
|
||
f"Параметр `sleep_time` должен быть не менее 5. Вы передали - {sleep_time}"
|
||
)
|
||
self.sleep_time = sleep_time
|
||
# проверяем переданный параметр способа сохранения капчи
|
||
if save_format in ["const", "temp"]:
|
||
self.save_format = save_format
|
||
else:
|
||
raise ValueError(
|
||
"\nПередан неверный формат сохранения файла изображения. "
|
||
f"\n\tВозможные варинты: `temp` и `const`. Вы передали - `{save_format}`"
|
||
"\nWrong `save_format` parameter. Valid params: `const` or `temp`."
|
||
f"\n\tYour param - `{save_format}`"
|
||
)
|
||
# Пайлоад для создания задачи
|
||
self.task_payload = {
|
||
"clientKey": anticaptcha_key,
|
||
"task": {"type": "SquareNetTask"},
|
||
"softId": app_key,
|
||
}
|
||
|
||
# задаём callbackUrl если передан
|
||
if callbackUrl:
|
||
self.task_payload.update({"callbackUrl": callbackUrl})
|
||
|
||
# отправляем запрос на результат решения капчи, если ещё капча не решена - ожидаем 5 сек
|
||
# если всё ок - идём дальше
|
||
self.result_payload = {"clientKey": anticaptcha_key}
|
||
|
||
def __enter__(self):
|
||
return self
|
||
|
||
def __exit__(self, exc_type, exc_value, traceback):
|
||
if exc_type:
|
||
return False
|
||
return True
|
||
|
||
def __image_temp_saver(self, content: bytes):
|
||
"""
|
||
Метод сохраняет файл изображения как временный и отправляет его сразу на сервер для расшифровки.
|
||
:return: Возвращает ID капчи
|
||
"""
|
||
# Создаём пайлоад, вводим ключ от сайта, выбираем метод ПОСТ и ждём ответа в JSON-формате
|
||
self.task_payload["task"].update(
|
||
{"body": base64.b64encode(content).decode("utf-8")}
|
||
)
|
||
# Отправляем на рукапча изображение капчи и другие парметры,
|
||
# в результате получаем JSON ответ с номером решаемой капчи и получая ответ - извлекаем номер
|
||
captcha_id = requests.post(create_task_url, json=self.task_payload).json()
|
||
return captcha_id
|
||
|
||
def __image_const_saver(self, content: bytes):
|
||
"""
|
||
Метод создаёт папку и сохраняет в неё изображение, затем передаёт его на расшифровку и удалет файл.
|
||
:return: Возвращает ID капчи
|
||
"""
|
||
img_path = "PythonAntiCaptchaImages"
|
||
|
||
if not os.path.exists(img_path):
|
||
os.mkdir(img_path)
|
||
|
||
# Высчитываем хэш изображения, для того что бы сохранить его под уникальным именем
|
||
image_hash = hashlib.sha224(content).hexdigest()
|
||
|
||
with open(
|
||
os.path.join(img_path, "im-{0}.png".format(image_hash)), "wb"
|
||
) as out_image:
|
||
out_image.write(content)
|
||
|
||
with open(
|
||
os.path.join(img_path, "im-{0}.png".format(image_hash)), "rb"
|
||
) as captcha_image:
|
||
# Добавляем в пайлоад картинку и отправляем
|
||
self.task_payload["task"].update(
|
||
{"body": base64.b64encode(captcha_image.read()).decode("utf-8")}
|
||
)
|
||
# Отправляем на антикапча изображение капчи и другие парметры,
|
||
# в результате получаем JSON ответ содержащий номер решаемой капчи
|
||
captcha_id = requests.post(create_task_url, json=self.task_payload).json()
|
||
|
||
# удаляем файл капчи
|
||
os.remove(os.path.join(img_path, "im-{0}.png".format(image_hash)))
|
||
|
||
return captcha_id
|
||
|
||
def __read_captcha_image_file(self, content: bytes, content_type: str = "file"):
|
||
"""
|
||
Функция отвечает за чтение уже сохранённого файла или файла в уодировке base64
|
||
:param content: Параметр строка-путь указывающий на изображение капчи для отправки её на сервер
|
||
:param content_type: Тип переданной переменной. Возможны варианты: `file` и `base64`
|
||
:return: Возвращает ID капчи
|
||
"""
|
||
try:
|
||
if content_type == "file":
|
||
with open(content, "rb") as captcha_image:
|
||
# Добавляем в пайлоад картинку и отправляем
|
||
self.task_payload["task"].update(
|
||
{"body": base64.b64encode(captcha_image.read()).decode("utf-8")}
|
||
)
|
||
elif content_type == "base64":
|
||
self.task_payload["task"].update({"body": content})
|
||
else:
|
||
raise ValueError(
|
||
"\nПередан неверный формат файла."
|
||
f"\n\tВозможные варинты: `file` и `base64`. Вы передали - `{content_type}`"
|
||
f"\nWrong `content_type` parameter. Valid params: `file` or `base64`."
|
||
f"\n\tYour param - `{content_type}`"
|
||
)
|
||
# Отправляем на антикапча изображение капчи и другие парметры,
|
||
# в результате получаем JSON ответ содержащий номер решаемой капчи
|
||
captcha_id = requests.post(create_task_url, json=self.task_payload).json()
|
||
except (IOError, FileNotFoundError) as err:
|
||
raise ReadError(err)
|
||
|
||
return captcha_id
|
||
|
||
# Работа с капчёй
|
||
def captcha_handler(
|
||
self,
|
||
objectName: str,
|
||
rowsCount: int,
|
||
columnsCount: int,
|
||
image_link: str = None,
|
||
image_file: str = None,
|
||
image_base64: str = None,
|
||
):
|
||
"""
|
||
Этот тип задачи берет вашу картинку, добавляет на нее сетку нужного размера и отдает работнику с требованием выбрать объекты нужного типа.
|
||
:param objectName: Имя объекта.
|
||
:param rowsCount: Кол-во строк. Min 2, max 5
|
||
:param columnsCount: Кол-во колонок. Min 2, max 5
|
||
|
||
:param image_link: Ссылка на изображение
|
||
:param image_file: Необязательный параметр, служит для открытия уже скачанных файлов изображений.
|
||
:param image_base64: Загрузка изображения в кодировке base64
|
||
|
||
:return: Возвращает весь ответ сервера JSON-строкой.
|
||
"""
|
||
# проверка параметров сетки
|
||
if not 2 <= rowsCount <= 5 or not 2 <= columnsCount <= 5:
|
||
raise ValueError(
|
||
"Параметр `rowsCount` и `columnsCount` должен быть больше 2 и меньше 5."
|
||
f"Вы передали - `rowsCount`:{rowsCount}, `columnsCount`:{columnsCount}"
|
||
)
|
||
|
||
# обновляем task_payload новыми данными для сетки разметки и названия предмета
|
||
self.task_payload["task"].update({"objectName": objectName})
|
||
self.task_payload["task"].update({"rowsCount": rowsCount})
|
||
self.task_payload["task"].update({"columnsCount": columnsCount})
|
||
|
||
# проводим действия над файлом(декодируем и передаём на сервер)
|
||
if image_file:
|
||
captcha_id = self.__read_captcha_image_file(image_file, content_type="file")
|
||
|
||
# проводим действия над файлом уже закодированном в base64(передаём на сервер)
|
||
elif image_base64:
|
||
captcha_id = self.__read_captcha_image_file(
|
||
image_base64, content_type="base64"
|
||
)
|
||
|
||
# проводим действия над ссылкой на файл(скачиваем, сохраняем и передаём на сервер)
|
||
elif image_link:
|
||
content = requests.get(image_link).content
|
||
# согласно значения переданного параметра выбираем функцию для сохранения изображения
|
||
if self.save_format == "const":
|
||
captcha_id = self.__image_const_saver(content)
|
||
elif self.save_format == "temp":
|
||
captcha_id = self.__image_temp_saver(content)
|
||
else:
|
||
raise ParamError(
|
||
additional_info="""You must set one of param - `image_file/image_base64/image_link`.\n
|
||
Вы должны задать один из параметров - `image_file/image_base64/image_link`."""
|
||
)
|
||
|
||
# Проверка статуса создания задачи, если создано без ошибок - извлекаем 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 aioSquareNetTextTask:
|
||
"""
|
||
Данный метод подходит для всинхронного решения капчи-изображение.
|
||
Подробней информацию смотрите в методе 'captcha_handler' и '__init__'
|
||
"""
|
||
|
||
def __init__(
|
||
self,
|
||
anticaptcha_key: str,
|
||
sleep_time: int = 5,
|
||
save_format: str = "temp",
|
||
callbackUrl: str = None,
|
||
):
|
||
"""
|
||
Инициализация нужных переменных, создание папки для изображений и кэша
|
||
После завершения работы - удаляются временные фалйы и папки
|
||
:param anticaptcha_key: АПИ ключ капчи из кабинета пользователя
|
||
:param sleep_time: Вермя ожидания решения капчи
|
||
:param save_format: Формат в котором будет сохраняться изображение, либо как временный файл - 'temp',
|
||
либо как обычное изображение в папку созданную библиотекой - 'const'.
|
||
:param callbackUrl: URL для решения капчи с ответом через callback
|
||
"""
|
||
if sleep_time < 5:
|
||
raise ValueError(
|
||
f"Параметр `sleep_time` должен быть не менее 5. Вы передали - {sleep_time}"
|
||
)
|
||
self.sleep_time = sleep_time
|
||
# проверяем переданный параметр способа сохранения капчи
|
||
if save_format in ["const", "temp"]:
|
||
self.save_format = save_format
|
||
else:
|
||
raise ValueError(
|
||
"\nПередан неверный формат сохранения файла изображения. "
|
||
f"\n\tВозможные варинты: `temp` и `const`. Вы передали - `{save_format}`"
|
||
"\nWrong `save_format` parameter. Valid params: `const` or `temp`."
|
||
f"\n\tYour param - `{save_format}`"
|
||
)
|
||
# Пайлоад для создания задачи
|
||
self.task_payload = {
|
||
"clientKey": anticaptcha_key,
|
||
"task": {"type": "SquareNetTask"},
|
||
"softId": app_key,
|
||
}
|
||
|
||
# задаём callbackUrl если передан
|
||
if callbackUrl:
|
||
self.task_payload.update({"callbackUrl": callbackUrl})
|
||
|
||
# отправляем запрос на результат решения капчи, если ещё капча не решена - ожидаем 5 сек
|
||
# если всё ок - идём дальше
|
||
self.result_payload = {"clientKey": anticaptcha_key}
|
||
|
||
def __enter__(self):
|
||
return self
|
||
|
||
def __exit__(self, exc_type, exc_value, traceback):
|
||
if exc_type:
|
||
return False
|
||
return True
|
||
|
||
async def __image_temp_saver(self, captcha_link: str):
|
||
"""
|
||
Метод сохраняет файл изображения как временный и отправляет его сразу на сервер для расшифровки.
|
||
:return: Возвращает ID капчи
|
||
"""
|
||
# Скачиваем капчу
|
||
async with aiohttp.ClientSession() as session:
|
||
async with session.get(captcha_link) as resp:
|
||
content = await resp.content.readany()
|
||
# Создаём пайлоад, вводим ключ от сайта, выбираем метод ПОСТ и ждём ответа в JSON-формате
|
||
self.task_payload["task"].update(
|
||
{"body": base64.b64encode(content).decode("utf-8")}
|
||
)
|
||
# Отправляем на рукапча изображение капчи и другие парметры,
|
||
# в результате получаем JSON ответ с номером решаемой капчи
|
||
async with aiohttp.ClientSession() as session:
|
||
async with session.post(create_task_url, json=self.task_payload) as resp:
|
||
return await resp.json()
|
||
|
||
async def __image_const_saver(self, captcha_link: str):
|
||
"""
|
||
Метод создаёт папку и сохраняет в неё изображение, затем передаёт его на расшифровку и удалет файл.
|
||
:return: Возвращает ID капчи
|
||
"""
|
||
img_path = "PythonAntiCaptchaImages"
|
||
|
||
if not os.path.exists(img_path):
|
||
os.mkdir(img_path)
|
||
|
||
# Скачиваем капчу
|
||
async with aiohttp.ClientSession() as session:
|
||
async with session.get(captcha_link) as resp:
|
||
content = await resp.content.readany()
|
||
|
||
# Высчитываем хэш изображения, для того что бы сохранить его под уникальным именем
|
||
image_hash = hashlib.sha224(content).hexdigest()
|
||
|
||
with open(
|
||
os.path.join(img_path, "im-{0}.png".format(image_hash)), "wb"
|
||
) as out_image:
|
||
out_image.write(content)
|
||
|
||
with open(
|
||
os.path.join(img_path, "im-{0}.png".format(image_hash)), "rb"
|
||
) as captcha_image:
|
||
# Добавляем в пайлоад картинку и отправляем
|
||
self.task_payload["task"].update(
|
||
{"body": base64.b64encode(captcha_image.read()).decode("utf-8")}
|
||
)
|
||
# Отправляем на антикапча изображение капчи и другие парметры,
|
||
# в результате получаем 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()
|
||
|
||
# удаляем файл капчи
|
||
os.remove(os.path.join(img_path, "im-{0}.png".format(image_hash)))
|
||
return captcha_id
|
||
|
||
async def __read_captcha_image_file(
|
||
self, content: bytes, content_type: str = "file"
|
||
):
|
||
"""
|
||
Функция отвечает за чтение уже сохранённого файла или файла в уодировке base64
|
||
:param content: Параметр строка-путь указывающий на изображение капчи для отправки её на сервер
|
||
:param content_type: Тип переданной переменной. Возможны варианты: `file` и `base64`
|
||
:return: Возвращает ID капчи
|
||
"""
|
||
try:
|
||
if content_type == "file":
|
||
with open(content, "rb") as captcha_image:
|
||
# Добавляем в пайлоад картинку и отправляем
|
||
self.task_payload["task"].update(
|
||
{"body": base64.b64encode(captcha_image.read()).decode("utf-8")}
|
||
)
|
||
elif content_type == "base64":
|
||
self.task_payload["task"].update({"body": content})
|
||
else:
|
||
raise ValueError(
|
||
"\nПередан неверный формат файла."
|
||
f"\n\tВозможные варинты: `file` и `base64`. Вы передали - `{content_type}`"
|
||
f"\nWrong `content_type` parameter. Valid params: `file` or `base64`."
|
||
f"\n\tYour param - `{content_type}`"
|
||
)
|
||
# Отправляем на антикапча изображение капчи и другие парметры,
|
||
# в результате получаем JSON ответ содержащий номер решаемой капчи
|
||
captcha_id = requests.post(create_task_url, json=self.task_payload).json()
|
||
except (IOError, FileNotFoundError) as err:
|
||
raise ReadError(err)
|
||
|
||
return captcha_id
|
||
|
||
# Работа с капчёй
|
||
async def captcha_handler(
|
||
self,
|
||
objectName: str,
|
||
rowsCount: int,
|
||
columnsCount: int,
|
||
image_link: str = None,
|
||
image_file: str = None,
|
||
image_base64: str = None,
|
||
):
|
||
"""
|
||
Этот тип задачи берет вашу картинку, добавляет на нее сетку нужного размера и отдает работнику с требованием выбрать объекты нужного типа.
|
||
:param objectName: Имя объекта.
|
||
:param rowsCount: Кол-во строк. Min 2, max 5
|
||
:param columnsCount: Кол-во колонок. Min 2, max 5
|
||
|
||
:param image_link: Ссылка на изображение
|
||
:param image_file: Необязательный параметр, служит для открытия уже скачанных файлов изображений.
|
||
:param image_base64: Загрузка изображения в кодировке base64
|
||
:return: Возвращает весь ответ сервера JSON-строкой.
|
||
"""
|
||
# проверка параметров сетки
|
||
if (2 < rowsCount < 5) and (2 < columnsCount < 5):
|
||
raise ValueError(
|
||
"Параметр `rowsCount` и `columnsCount` должен быть больше 2 и меньше 5."
|
||
f"Вы передали - `rowsCount`:{rowsCount}, `columnsCount`:{columnsCount}"
|
||
)
|
||
|
||
# обновляем task_payload новыми данными для сетки разметки и названия предмета
|
||
self.task_payload["task"].update({"objectName": objectName})
|
||
self.task_payload["task"].update({"rowsCount": rowsCount})
|
||
self.task_payload["task"].update({"columnsCount": columnsCount})
|
||
# если был передан линк на локальный скачаный файл
|
||
if image_file:
|
||
captcha_id = await self.__read_captcha_image_file(
|
||
image_file, content_type="file"
|
||
)
|
||
elif image_base64:
|
||
captcha_id = await self.__read_captcha_image_file(
|
||
image_base64, content_type="base64"
|
||
)
|
||
elif image_link:
|
||
# согласно значения переданного параметра выбираем функцию для сохранения изображения
|
||
if self.save_format == "const":
|
||
captcha_id = await self.__image_const_saver(image_link)
|
||
elif self.save_format == "temp":
|
||
captcha_id = await self.__image_temp_saver(image_link)
|
||
else:
|
||
raise ParamError(
|
||
additional_info="""You must set one of param - `image_file/image_base64/image_link`.\n
|
||
Вы должны задать один из параметров - `image_file/image_base64/image_link`."""
|
||
)
|
||
|
||
# Проверка статуса создания задачи, если создано без ошибок - извлекаем 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
|
||
)
|