James R. Barlow
3d291e72c0
Fix the 13 errors that surfaced under mypy --check-untyped-defs so the flag can be turned on permanently in pyproject.toml, and drop the advisory exit-0 wrapper on the mypy pre-commit hook now that the tree is clean. - _plugin_manager: rename colliding loop vars (module/name were reused with conflicting types) and guard spec/spec.loader from spec_from_file_location; call __init__ via the class in __setstate__. - __main__: pass Verbosity(options.verbose), not a bare int. - subprocess/_check: widen package to str | Mapping[str, str] to match _error_trailer's existing per-platform handling. - optimize.main: annotate the standalone PdfContext(..., None, None) that only ever reads context.options.
90 lines
2.5 KiB
Python
Executable File
90 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# SPDX-FileCopyrightText: 2022 James R. Barlow
|
|
# SPDX-License-Identifier: MPL-2.0
|
|
|
|
"""ocrmypdf command line entrypoint."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import multiprocessing
|
|
import os
|
|
import signal
|
|
import sys
|
|
from contextlib import suppress
|
|
|
|
from ocrmypdf import __version__
|
|
from ocrmypdf._pipelines.ocr import run_pipeline_cli
|
|
from ocrmypdf._validation import check_options
|
|
from ocrmypdf.api import Verbosity, configure_logging, configure_stdout_protection
|
|
from ocrmypdf.cli import get_options_and_plugins
|
|
from ocrmypdf.exceptions import (
|
|
BadArgsError,
|
|
ExitCode,
|
|
InputFileError,
|
|
MissingDependencyError,
|
|
)
|
|
|
|
log = logging.getLogger('ocrmypdf')
|
|
|
|
|
|
def sigbus(*args):
|
|
"""Handle SIGBUS signals.
|
|
|
|
pikepdf, depending on configuration, may use mmap so SIGBUS is a
|
|
possibility.
|
|
"""
|
|
raise InputFileError("Lost access to the input file")
|
|
|
|
|
|
def run(args=None):
|
|
"""Run the ocrmypdf command line interface."""
|
|
# Protect the real stdout before loading plugins or starting any worker
|
|
# processes/threads, so that only our final PDF output can reach it and
|
|
# stray writes from plugins or libraries are diverted to stderr.
|
|
configure_stdout_protection()
|
|
|
|
options, plugin_manager = get_options_and_plugins(args=args)
|
|
|
|
with suppress(AttributeError, PermissionError):
|
|
os.nice(5)
|
|
|
|
verbosity = Verbosity(options.verbose)
|
|
if not os.isatty(sys.stderr.fileno()):
|
|
options.progress_bar = False
|
|
if options.quiet:
|
|
verbosity = Verbosity.quiet
|
|
options.progress_bar = False
|
|
configure_logging(
|
|
verbosity,
|
|
progress_bar_friendly=options.progress_bar,
|
|
manage_root_logger=True,
|
|
plugin_manager=plugin_manager,
|
|
)
|
|
log.debug('ocrmypdf %s', __version__)
|
|
try:
|
|
check_options(options, plugin_manager)
|
|
except ValueError as e:
|
|
log.error(e)
|
|
return ExitCode.bad_args
|
|
except BadArgsError as e:
|
|
log.error(e)
|
|
return e.exit_code
|
|
except MissingDependencyError as e:
|
|
log.error(e)
|
|
return ExitCode.missing_dependency
|
|
|
|
with suppress(AttributeError, OSError):
|
|
signal.signal(signal.SIGBUS, sigbus)
|
|
|
|
result = run_pipeline_cli(options=options, plugin_manager=plugin_manager)
|
|
return result
|
|
|
|
|
|
if __name__ == '__main__':
|
|
multiprocessing.freeze_support()
|
|
if sys.platform not in ('win32', 'darwin'):
|
|
with suppress(RuntimeError):
|
|
multiprocessing.set_start_method('forkserver')
|
|
sys.exit(run())
|