Обновлено описание для ImageToText и небольшие правки кода, теперь

функция возвращает напрмую ответ сервера.
Небольшие синтаксические правки в NoCaptchaTask
This commit is contained in:
Andrei 2017-10-22 20:18:37 +03:00
parent bfc791aaa2
commit c9113b0e05
4 changed files with 37 additions and 33 deletions

1
.gitignore vendored
View File

@ -5,3 +5,4 @@
.idea/misc.xml
.idea/modules.xml
.idea/workspace.xml
/test.py

View File

@ -2,6 +2,8 @@
Python 3 library for AntiCaptcha.
Tested on UNIX based OS.
Библиотека предназначена для разрабаотчиков ПО и служит для облегчения работы с API сервиса AntiCaptcha.
## How to install? Как установить?

View File

@ -11,20 +11,20 @@ from .config import create_task_url, get_result_url, app_key
class ImageToTextTask:
'''
Данный метод подходит как для загрузки и решения обычной капчи
так и для большой капчи.
Требуется передать API ключ сайта, ссылку на изображение и,по желанию, время ожидания решения капчи
Подробней информацию смотрите в методе 'captcha_handler'
Данный метод подходит для решения капчи-изображение.
Подробней информацию смотрите в методе 'captcha_handler' и '__init__'
'''
def __init__(self, anticaptcha_key, sleep_time=5, save_format = 'temp', language = 'en',**kwargs):
'''
Инициализация нужных переменных, создание папки для изображений и кэша
После завершения работы - удаляются временные фалйы и папки
:param rucaptcha_key: АПИ ключ капчи из кабинета пользователя
:param anticaptcha_key: АПИ ключ капчи из кабинета пользователя
:param sleep_time: Вермя ожидания решения капчи
:param save_format: Формат в котором будет сохраняться изображение, либо как временный файл - 'temp',
либо как обычное изображение в папку созданную библиотекой - 'const'.
:param language: Язык капчи
:param **kwargs: За подробной информацией обратитесь к документации на сайте anticaptcha.
'''
self.ANTICAPTCHA_KEY = anticaptcha_key
@ -39,6 +39,11 @@ class ImageToTextTask:
},
"languagePool": language
}
# отправляем запрос на результат решения капчи, если ещё капча не решена - ожидаем 5 сек
# если всё ок - идём дальше
self.result_payload = {"clientKey": self.ANTICAPTCHA_KEY}
# Если переданы ещё параметры - вносим их в payload
if kwargs:
for key in kwargs:
@ -49,15 +54,13 @@ class ImageToTextTask:
Метод сохраняет файл изображения как временный и отправляет его сразу на сервер для расшифровки.
:return: Возвращает ID капчи
'''
with tempfile.NamedTemporaryFile(suffix='.png') as out:
out.write(content)
with open(out.name, 'rb') as captcha_image:
with tempfile.NamedTemporaryFile(suffix='.png') as captcha_image:
captcha_image.write(content)
# Создаём пайлоад, вводим ключ от сайта, выбираем метод ПОСТ и ждём ответа в JSON-формате
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())
captcha_id = (requests.post(create_task_url, json = self.task_payload).json())
return captcha_id
def image_const_saver(self, content):
@ -80,11 +83,10 @@ class ImageToTextTask:
# Добавляем в пайлоад картинку и отправляем
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())
# в результате получаем 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
@ -95,7 +97,7 @@ class ImageToTextTask:
Метод получает от вас ссылку на изображение, скачивает его, отправляет изображение на сервер
RuCaptcha, дожидается решения капчи и вовзращает вам результат
:param captcha_link: Ссылка на изображение
:return: Возвращает список из 2 элементов: 1. Решённая капча; 2. Весь ответ сервера.
:return: Возвращает весь ответ сервера.
'''
content = requests.get(captcha_link).content
@ -109,29 +111,27 @@ class ImageToTextTask:
return """Wrong 'save_format' parameter. Valid formats: 'const' or 'temp'.\n
Неправильный 'save_format' параметр. Возможные форматы: 'const' или 'temp'."""
# Проверка статуса создания задачи, если создано без ошибок - извлекаем ID задачи, иначе возвращаем тело ошибки
# Проверка статуса создания задачи, если создано без ошибок - извлекаем ID задачи, иначе возвращаем ответ сервера
if captcha_id['errorId'] == 0:
captcha_id = captcha_id["taskId"]
self.result_payload.update({"taskId": captcha_id})
else:
return captcha_id
# Ожидаем решения капчи
time.sleep(self.sleep_time)
while True:
# отправляем запрос на результат решения капчи, если ещё капча не решена - ожидаем 5 сек
# если всё ок - идём дальше
result_payload = {"clientKey": self.ANTICAPTCHA_KEY,
"taskId": captcha_id
}
# отправляем запрос на результат решения капчи, если не решена ожидаем
captcha_response = requests.post(get_result_url, json = result_payload)
captcha_response = requests.post(get_result_url, json = self.result_payload)
# Если ошибки нет - проверяем статус капчи
if captcha_response.json()['errorId'] == 0:
# Если капча ещё не готова- ожидаем
if captcha_response.json()["status"] == "processing":
time.sleep(self.sleep_time)
elif captcha_response.json()["status"] == "ready":
return captcha_response.json()["solution"]["text"], captcha_response.json()
# если уже решена - возвращаем ответ сервера
else:
return captcha_response.json()
# Иначе возвращаем ответ сервера
else:
return captcha_response.json()

View File

@ -1,6 +1,6 @@
import requests
from .config import create_tast_url, get_result_url, app_key
from .config import create_task_url, get_result_url, app_key
#TODO from .errors import AntiCaptchaError
@ -35,8 +35,9 @@ class NoCaptchaTask:
if kwargs:
for key in kwargs:
self.task_payload['task'].update(kwars) # is it error?
self.task_payload['task'].update(kwargs) # is it error?
def captcha_handler(self):
#TODO
pass