9564db398f
Remove support for quoted entried. Support unquoted entries. Support custom location for the config files. Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
120 lines
3.7 KiB
Python
120 lines
3.7 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
import logging
|
|
import shutil
|
|
|
|
import sh
|
|
|
|
from django.utils.encoding import force_text
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from mayan.apps.storage.utils import TemporaryFile
|
|
|
|
from ..classes import OCRBackendBase
|
|
from ..exceptions import OCRError
|
|
from ..settings import setting_ocr_backend_arguments
|
|
|
|
from .literals import DEFAULT_TESSERACT_BINARY_PATH, DEFAULT_TESSERACT_TIMEOUT
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Tesseract(OCRBackendBase):
|
|
def __init__(self, *args, **kwargs):
|
|
auto_initialize = kwargs.pop('auto_initialize', True)
|
|
super(Tesseract, self).__init__(*args, **kwargs)
|
|
|
|
if auto_initialize:
|
|
self.initialize()
|
|
|
|
def execute(self, *args, **kwargs):
|
|
"""
|
|
Execute the command line binary of tesseract
|
|
"""
|
|
super(Tesseract, self).execute(*args, **kwargs)
|
|
|
|
if self.command_tesseract:
|
|
image = self.converter.get_page()
|
|
|
|
try:
|
|
temporary_image_file = TemporaryFile()
|
|
shutil.copyfileobj(image, temporary_image_file)
|
|
temporary_image_file.seek(0)
|
|
|
|
arguments = ['-', '-']
|
|
|
|
keyword_arguments = {
|
|
'_in': temporary_image_file,
|
|
'_timeout': self.command_timeout
|
|
}
|
|
|
|
if self.language:
|
|
keyword_arguments['l'] = self.language
|
|
|
|
try:
|
|
|
|
result = self.command_tesseract(
|
|
*arguments, **keyword_arguments
|
|
)
|
|
return force_text(result.stdout)
|
|
except Exception as exception:
|
|
error_message = (
|
|
'Exception calling Tesseract with language option: {}; {}'
|
|
).format(self.language, exception)
|
|
|
|
if self.language not in self.languages:
|
|
error_message = (
|
|
'{}\nThe requested OCR language "{}" is not '
|
|
'available and needs to be installed.\n'
|
|
).format(
|
|
error_message, self.language
|
|
)
|
|
|
|
logger.error(error_message)
|
|
raise OCRError(error_message)
|
|
else:
|
|
return result
|
|
finally:
|
|
temporary_image_file.close()
|
|
|
|
def initialize(self):
|
|
self.languages = ()
|
|
|
|
self.read_settings()
|
|
|
|
try:
|
|
self.command_tesseract = sh.Command(path=self.tesseract_binary_path)
|
|
except sh.CommandNotFound:
|
|
self.command_tesseract = None
|
|
raise OCRError(
|
|
_('Tesseract OCR not found.')
|
|
)
|
|
else:
|
|
# Get version
|
|
result = self.command_tesseract(v=True)
|
|
logger.debug('Tesseract version: %s', result.stdout)
|
|
|
|
# Get languages
|
|
result = self.command_tesseract(list_langs=True)
|
|
# Sample output format
|
|
# List of available languages (3):
|
|
# deu
|
|
# eng
|
|
# osd
|
|
# <- empty line
|
|
|
|
# Extaction: strip last line, split by newline, discard the first
|
|
# line
|
|
self.languages = force_text(result.stdout).strip().split('\n')[1:]
|
|
|
|
logger.debug('Available languages: %s', ', '.join(self.languages))
|
|
|
|
def read_settings(self):
|
|
self.tesseract_binary_path = setting_ocr_backend_arguments.value.get(
|
|
'tesseract_path', DEFAULT_TESSERACT_BINARY_PATH
|
|
)
|
|
|
|
self.command_timeout = setting_ocr_backend_arguments.value.get(
|
|
'timeout', DEFAULT_TESSERACT_TIMEOUT
|
|
)
|