from __future__ import unicode_literals

import logging

from PIL import Image
import pytesseract

from ..classes import OCRBackendBase
from ..exceptions import OCRError
from ..settings import setting_tesseract_path

logger = logging.getLogger(__name__)


class Tesseract(OCRBackendBase):
    def execute(self, *args, **kwargs):
        """
        Execute the command line binary of tesseract
        """
        super(Tesseract, self).execute(*args, **kwargs)

        # TODO: pass tesseract binary path to the pytesseract
        image = Image.open(self.converter.get_page())
        try:
            result = pytesseract.image_to_string(image=image, lang=self.language)
            # If tesseract gives an error with a language parameter
            # re-run it with no language parameter
        except Exception as exception:
            logger.error('Exception calling pytesseract with language option: %s; %s', self.language, exception)
            try:
                result = pytesseract.image_to_string(image=image)
            except Exception as exception:
                logger.debug('Exception calling pytesseract without a language option; %s', exception)
                raise OCRError

        return result