85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
# © 2015 James R. Barlow: github.com/jbarlow83
|
|
|
|
from subprocess import CalledProcessError, check_output, STDOUT, check_call
|
|
import sys
|
|
import os
|
|
|
|
from . import ExitCode, get_program
|
|
|
|
|
|
def check(input_file, log):
|
|
args_qpdf = [
|
|
get_program('qpdf'),
|
|
'--check',
|
|
input_file
|
|
]
|
|
|
|
try:
|
|
check_output(args_qpdf, stderr=STDOUT, universal_newlines=True)
|
|
except CalledProcessError as e:
|
|
if e.returncode == 2:
|
|
print("{0}: not a valid PDF, and could not repair it.".format(
|
|
input_file))
|
|
print("Details:")
|
|
print(e.output)
|
|
elif e.returncode == 3:
|
|
log.info("qpdf --check returned warnings:")
|
|
log.info(e.output)
|
|
else:
|
|
print(e.output)
|
|
return False
|
|
return True
|
|
|
|
|
|
def repair(input_file, output_file, log):
|
|
args_qpdf = [
|
|
get_program('qpdf'), input_file, output_file
|
|
]
|
|
try:
|
|
check_output(args_qpdf, stderr=STDOUT, universal_newlines=True)
|
|
except CalledProcessError as e:
|
|
if e.returncode == 3 and e.output.find("operation succeeded"):
|
|
log.debug('qpdf found and fixed errors:')
|
|
log.debug(e.output)
|
|
print(e.output)
|
|
return
|
|
|
|
if e.returncode == 2 and e.output.find("invalid password"):
|
|
print("{0}: this PDF is password-protected - password must "
|
|
"be removed for OCR".format(input_file))
|
|
sys.exit(ExitCode.input_file)
|
|
elif e.returncode == 2:
|
|
print("{0}: not a valid PDF, and could not repair it.".format(
|
|
input_file))
|
|
print("Details:")
|
|
print(e.output)
|
|
sys.exit(ExitCode.input_file)
|
|
else:
|
|
print("{0}: unknown error".format(
|
|
input_file))
|
|
print(e.output)
|
|
sys.exit(ExitCode.unknown)
|
|
|
|
|
|
def get_npages(input_file):
|
|
pages = check_output(
|
|
[get_program('qpdf'), '--show-npages', input_file],
|
|
universal_newlines=True, close_fds=True)
|
|
return int(pages)
|
|
|
|
|
|
def split_pages(input_file, work_folder, npages):
|
|
"""Split multipage PDF into individual pages.
|
|
|
|
Incredibly enough, this multiple process approach is about 70 times
|
|
faster than using Ghostscript.
|
|
"""
|
|
for n in range(int(npages)):
|
|
args_qpdf = [
|
|
get_program('qpdf'), input_file,
|
|
'--pages', input_file, '{0}'.format(n + 1), '--',
|
|
os.path.join(work_folder, '{0:06d}.page.pdf'.format(n + 1))
|
|
]
|
|
check_call(args_qpdf)
|