calebasse/calebasse/pdftk.py @ fc904112
| fa500e83 | Benjamin Dauvergne | import os
|
|
| 09630453 | Benjamin Dauvergne | import subprocess
|
|
import tempfile
|
|||
class PdfTk(object):
|
|||
| fa500e83 | Benjamin Dauvergne | def __init__(self, pdftk_path=None, prefix='tmp'):
|
|
| 09630453 | Benjamin Dauvergne | self._pdftk_path = pdftk_path
|
|
| fa500e83 | Benjamin Dauvergne | self.prefix = prefix
|
|
| 09630453 | Benjamin Dauvergne | ||
@property
|
|||
def pdftk_path(self):
|
|||
return self._pdftk_path or '/usr/bin/pdftk'
|
|||
| fa500e83 | Benjamin Dauvergne | def do(self, args, wait=True):
|
|
| 09630453 | Benjamin Dauvergne | args = [self.pdftk_path] + args
|
|
proc = subprocess.Popen(args)
|
|||
| fa500e83 | Benjamin Dauvergne | if wait:
|
|
return proc.wait()
|
|||
else:
|
|||
return proc
|
|||
| 09630453 | Benjamin Dauvergne | ||
| fa500e83 | Benjamin Dauvergne | def concat(self, input_files, output_file, wait=True):
|
|
| 7c757f94 | Benjamin Dauvergne | args = input_files + ['cat', 'output', output_file, 'compress', 'flatten']
|
|
return self.do(args, wait=wait)
|
|||
def background(self, input_file, background_file, output_file, wait=True):
|
|||
args = [input_file, 'background', background_file, 'output', output_file, 'compress']
|
|||
| fa500e83 | Benjamin Dauvergne | return self.do(args, wait=wait)
|
|
| 09630453 | Benjamin Dauvergne | ||
if __name__ == '__main__':
|
|||
import sys
|
|||
pdftk = PdfTk()
|
|||
if sys.argv[1] == 'concat':
|
|||
pdftk.concat(sys.argv[2].split(','), sys.argv[3])
|
|||
| 775dcb4b | Benjamin Dauvergne | elif sys.argv[1] == 'background':
|
|
pdftk.background(*sys.argv[2:5])
|
|||
| 09630453 | Benjamin Dauvergne | else:
|
|
raise RuntimeError('Unknown command %s', sys.argv)
|