Rooba
/
python3-anticaptcha
mirror of https://github.com/AndreiDrang/python3-anticaptcha.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add support to work directly with base64 image

This commit is contained in:
Wanderson Ferreira 2018-05-12 13:52:00 -03:00
parent 2e9e7d4dbf
commit e1f0dac496
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
python3_anticaptcha/ImageToTextTask.py
View File

@ -92,16 +92,21 @@ class ImageToTextTask:
return captcha_id
def read_captcha_image_file(self, content):
def read_captcha_image_file(self, content, content_type="file"):
"""
Функция отвечает за чтение уже сохранённого файла
:param content: Параметр строка-путь указывающий на изображение капчи для отправки её на сервер
: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.decode('utf-8')})
else:
raise Exception
# Отправляем на антикапча изображение капчи и другие парметры,
# в результате получаем JSON ответ содержащий номер решаемой капчи
captcha_id = requests.post(create_task_url, json=self.task_payload).json()
@ -110,9 +115,8 @@ class ImageToTextTask:
return captcha_id
# Работа с капчёй
def captcha_handler(self, captcha_link = None, captcha_file = None, **kwargs):
def captcha_handler(self, captcha_link=None, captcha_file=None, captcha_base64=None, **kwargs):
'''
Метод получает от вас ссылку на изображение, скачивает его, отправляет изображение на сервер
RuCaptcha, дожидается решения капчи и вовзращает вам результат
@ -121,7 +125,9 @@ class ImageToTextTask:
:return: Возвращает весь ответ сервера JSON-строкой.
'''
if captcha_file:
captcha_id = self.read_captcha_image_file(captcha_file)
captcha_id = self.read_captcha_image_file(captcha_file, content_type="file")
elif captcha_base64:
captcha_id = self.read_captcha_image_file(captcha_base64, content_type="base64")
elif captcha_link:
content = requests.get(captcha_link, **kwargs).content
# согласно значения переданного параметра выбираем функцию для сохранения изображения