mirror of
https://github.com/ocrmypdf/OCRmyPDF.git
synced 2025-11-02 10:50:29 +00:00
76 lines
2.1 KiB
Python
76 lines
2.1 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
|
|
|
|
|
|
def check(input_file, log):
|
|
args_qpdf = [
|
|
'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 = [
|
|
'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:
|
|
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(
|
|
['qpdf', '--show-npages', input_file],
|
|
universal_newlines=True, close_fds=True)
|
|
return int(pages)
|
|
|
|
|
|
def split_pages(input_file, work_folder, npages):
|
|
for n in range(int(npages)):
|
|
args_qpdf = [
|
|
'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)
|