addon/platformcode/recaptcha.py

# -*- coding: utf-8 -*-
import random
import time
from threading import Thread

import channelselector
import xbmcgui
from core import httptools, support
from core import filetools
from platformcode import config, platformtools
from platformcode import logger
from lib.librecaptcha.recaptcha import ReCaptcha, Solver, DynamicSolver, MultiCaptchaSolver, Solution, \
    ImageGridChallenge

lang = 'it'
temp_dir = config.get_temp_file('reCAPTCHA/')
tiles_pos = (75+390, 90+40)
grid_width = 450
tiles_texture_focus = 'white.png'
tiles_texture_checked = 'Controls/check_mark.png'
empty_image = 'https://upload.wikimedia.org/wikipedia/commons/4/49/A_black_image.jpg'

TITLE = 10
PANEL = 11
IMAGE = 12
CONTROL = 1

OK = 21
CANCEL = 22
# RELOAD = 23


def get_temp():
    if not filetools.isdir(temp_dir):
        filetools.mkdir(temp_dir)
    return temp_dir + str(random.randint(1, 1000)) + '.png'


class Kodi:
    def __init__(self, key, referer):
        filetools.rmdirtree(temp_dir)
        self.rc = ReCaptcha(
            api_key=key,
            site_url=referer,
            user_agent=httptools.get_user_agent(),
            lang = lang
        )

    def run(self):
        result = self.rc.first_solver()
        while not isinstance(result, str) and result is not False:
            solution = self.run_solver(result)
            if solution:
                result = self.rc.send_solution(solution)
                logger.debug(result)
            else:
                return False
        platformtools.dialog_notification("Captcha corretto", "Verifica conclusa")
        return result

    def run_solver(self, solver: Solver) -> Solution:
        selected_solver = {
            DynamicSolver: DynamicKodi,
            MultiCaptchaSolver: MultiCaptchaKodi,
        }[type(solver)]("Recaptcha.xml", config.get_runtime_path())
        selected_solver.solver = solver
        return selected_solver.run()


class SolverKodi(xbmcgui.WindowXMLDialog):
    def __init__(self, *args, **kwargs):
        self.goal = ""
        self.closed = False
        self.result = None
        self.image_path = ''
        self.indices = {}
        self.num_rows = 3
        self.num_columns = 3
        self.num_tiles = 9
        logger.debug()

    def show_image(self, image, goal):
        self.image_path = get_temp()
        filetools.write(self.image_path, image)
        self.goal = goal.replace('<strong>', '[B]').replace('</strong>', '[/B]')
        self.doModal()

    def onInit(self):
        logger.debug(self.image_path)
        items=[]
        self.getControl(IMAGE).setImage(self.image_path, False)
        self.getControl(TITLE).setLabel(self.goal)

        for x in range(self.num_tiles):
            item = xbmcgui.ListItem('')
            item.setProperty('selected', 'false')
            items.append(item)
        self.getControl(PANEL).reset()
        self.getControl(PANEL).addItems(items)


class MultiCaptchaKodi(SolverKodi):
    """
    multicaptcha challenges present you with one large image split into a grid of tiles and ask you to select the tiles that contain a given object.
    Occasionally, the image will not contain the object, but rather something that looks similar.
    It is possible to select no tiles in this case, but reCAPTCHA may have been fooled by the similar-looking object and would reject a selection of no tiles.
    """
    def run(self):
        result = self.solver.first_challenge()
        while not isinstance(result, Solution):
            if not isinstance(result, ImageGridChallenge):
                raise TypeError("Unexpected type: {}".format(type(result)))
            indices = self.handle_challenge(result)
            if self.closed:
                return False
            result = self.solver.select_indices(indices)
        return result

    def handle_challenge(self, challenge: ImageGridChallenge):
        goal = challenge.goal.plain
        self.num_rows = challenge.dimensions.rows
        self.num_columns = challenge.dimensions.columns
        logger.debug('RIGHE',self.num_rows, 'COLONNE',self.num_columns)

        self.num_tiles = challenge.dimensions.count
        image = challenge.image
        self.show_image(image, goal)
        if self.closed:
            return False
        return self.result

    def onClick(self, control):
        if control == CANCEL:
            self.closed = True
            self.close()

        # elif control == RELOAD:
        #     self.closed = True
        #     self.close()

        elif control == OK:
            self.result = [int(k) for k in range(self.num_tiles) if self.indices.get(k, False)]
            self.close()
        else:
            item = self.getControl(PANEL)
            index = item.getSelectedPosition()
            selected = True if item.getSelectedItem().getProperty('selected') == 'false' else False
            item.getSelectedItem().setProperty('selected', str(selected).lower())
            self.indices[index] = selected


class DynamicKodi(SolverKodi):
    """
    dynamic challenges present you with a grid of different images and ask you to select the images that match the given description.
    Each time you click an image, a new one takes its place. Usually, three images from the initial set match the description,
    and at least one of the replacement images does as well.
    """
    def run(self):
        challenge = self.solver.get_challenge()
        image = challenge.image
        goal = challenge.goal.raw
        self.num_rows = challenge.dimensions.rows
        self.num_columns = challenge.dimensions.columns
        self.num_tiles = challenge.dimensions.count
        logger.debug('RIGHE',self.num_rows, 'COLONNE',self.num_columns)

        self.show_image(image, goal)
        if self.closed:
            return False
        return self.result

    def changeTile(self, item, path, delay):
        cur_delay = delay
        while cur_delay > 0:
            # todo: show time
            item.setLabel('{:.1f}'.format(cur_delay))
            time.sleep(0.1)
            cur_delay -= 0.1
        item.setLabel('')
        item.setArt({'image': path})

    def onClick(self, control):
        if control == CANCEL:
            self.closed = True
            self.close()

        # elif control == RELOAD:
        #     self.result = None
        #     self.close()

        elif control == OK:
            self.result = self.solver.finish()
            self.close()
        else:
            panel = self.getControl(PANEL)
            item = panel.getSelectedItem()
            if not item.getLabel():
                index = panel.getSelectedPosition()
                tile = self.solver.select_tile(index)
                path = get_temp()
                filetools.write(path, tile.image)
                Thread(target=self.changeTile, args=(item, path, tile.delay)).start()