From 486f73d5d6cc2887190424d17944a7e5ba8cc091 Mon Sep 17 00:00:00 2001 From: "James R. Barlow" Date: Wed, 15 May 2019 02:28:13 -0700 Subject: [PATCH] Remove custom logger --- src/ocrmypdf/_jobcontext.py | 67 +--------------------------------- src/ocrmypdf/exec/tesseract.py | 7 +++- tests/test_main.py | 3 +- 3 files changed, 8 insertions(+), 69 deletions(-) diff --git a/src/ocrmypdf/_jobcontext.py b/src/ocrmypdf/_jobcontext.py index 9a0a1eb2..f5ac6978 100644 --- a/src/ocrmypdf/_jobcontext.py +++ b/src/ocrmypdf/_jobcontext.py @@ -15,16 +15,12 @@ # You should have received a copy of the GNU General Public License # along with OCRmyPDF. If not, see . +import logging import shutil import sys import os from contextlib import suppress -ERROR = 40 -WARN = 30 -INFO = 20 -DEBUG = 10 - class PDFContext: """Holds our context for a particular run of the pipeline""" @@ -78,63 +74,4 @@ def cleanup_working_files(work_folder, options): def get_logger(options=None, prefix=''): - level = ERROR # TODO: add option - if options is not None: - if options.quiet or options.output_file == '-' or options.sidecar == '-': - return NullLogger() - if options.verbose == 0: - level = ERROR - elif options.verbose == 1: - level = WARN - elif options.verbose == 2: - level = INFO - elif options.verbose >= 3: - level = DEBUG - - return Logger(prefix, level) - - -class Logger: - def __init__(self, prefix, level=DEBUG): - self.prefix = prefix - self.level = level - - def debug(self, *args, **kwargs): - if self.level <= DEBUG: - print('DEBUG', self.prefix, end='', file=sys.stderr) - print(*args, file=sys.stderr, **kwargs) - - def info(self, *args, **kwargs): - if self.level <= INFO: - print('INFO', self.prefix, end='', file=sys.stderr) - print(*args, file=sys.stderr, **kwargs) - - def warning(self, *args, **kwargs): - self.warn(*args, **kwargs) - - def warn(self, *args, **kwargs): - if self.level <= WARN: - print('WARN', self.prefix, end='', file=sys.stderr) - print(*args, file=sys.stderr, **kwargs) - - def error(self, *args, **kwargs): - if self.level <= ERROR: - print('ERROR', self.prefix, end='', file=sys.stderr) - print(*args, file=sys.stderr) - - -class NullLogger: - def debug(self, *args, **kwargs): - pass - - def info(self, *args, **kwargs): - pass - - def warning(self, *args, **kwargs): - pass - - def warn(self, *args, **kwargs): - pass - - def error(self, *args, **kwargs): - pass + return logging.getLogger(prefix) diff --git a/src/ocrmypdf/exec/tesseract.py b/src/ocrmypdf/exec/tesseract.py index 84a728af..110c49f3 100644 --- a/src/ocrmypdf/exec/tesseract.py +++ b/src/ocrmypdf/exec/tesseract.py @@ -33,7 +33,11 @@ from subprocess import ( from textwrap import dedent from . import get_version -from ..exceptions import MissingDependencyError, TesseractConfigError, SubprocessOutputError +from ..exceptions import ( + MissingDependencyError, + TesseractConfigError, + SubprocessOutputError, +) from ..helpers import page_number OrientationConfidence = namedtuple('OrientationConfidence', ('angle', 'confidence')) @@ -342,7 +346,6 @@ def generate_pdf( # to the number of order parameters here args_tesseract.extend([input_image, prefix, 'pdf', 'txt'] + tessconfig) - try: log.debug(args_tesseract) stdout = check_output(args_tesseract, stderr=STDOUT, timeout=timeout) diff --git a/tests/test_main.py b/tests/test_main.py index 9461503f..2165fc78 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1014,8 +1014,7 @@ def test_bad_utf8(spoof_tess_bad_utf8, renderer, resources, no_outpdf): assert out == '', "stdout not clean" assert p.returncode != 0 assert 'not utf-8' in err, "should whine about utf-8" - # TODO: find out why this should be tested - # assert '\\x96' in err, 'should repeat backslash encoded output' + assert '\\x96' in err, 'should repeat backslash encoded output' @pytest.mark.skipif(