Skip to content

Instantly share code, notes, and snippets.

@ymotongpoo
Created August 29, 2011 17:01
Show Gist options
  • Save ymotongpoo/1178840 to your computer and use it in GitHub Desktop.
Save ymotongpoo/1178840 to your computer and use it in GitHub Desktop.
concat multiple pdf files
# -*- coding: utf-8 -*-
from pyPdf import PdfFileWriter, PdfFileReader
def main(output_file, input_files):
print "concat all files:"
output = PdfFileWriter()
total_pages = 0
for f in input_files:
# expect filename as "*.pdf"
if f[-4:] != ".pdf":
print "skipped file: ", f
continue
else:
input = PdfFileReader(file(f, 'rb'))
num_pages = input.getNumPages()
total_pages += num_pages
print f, "->", str(num_pages) + "pages"
for i in xrange(0, num_pages):
output.addPage(input.getPage(i))
outputStream = file(output_file, 'wb')
output.write(outputStream)
print str(total_pages) + "pages written"
outputStream.close()
if __name__ == '__main__':
import argparse
description = "concatinate input PDF files"
parser = argparse.ArgumentParser(description=description)
parser.add_argument("-o", "--output", default="output.pdf")
parser.add_argument("-i", "--input", nargs='*', required=True)
parser.add_argument("-v", "--version", action='version',
version="%(prog)s 1.0")
args = vars(parser.parse_args())
print "****** start concatination ******"
main(args['output'], args['input'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment