Script for adding picture pdf to a longer document

addcat.py

"""Split up a pdf, add an image, recombine pdf"""

from pyPdf import PdfFileWriter, PdfFileReader
from random import random
import argparse

def main():
    parser = argparse.ArgumentParser(description = """Generate new pdf file
    based on length of existing pdf""")
    parser.add_argument('input_paper', help = """Input pdf of interest""")
    parser.add_argument('image', help = """Image to be added""")
    parser.add_argument('output', help = """Name of output file""")
    args = parser.parse_args()

    inputpdf = PdfFileReader(file(args.input_paper, "rb"))
    inputpdf2 = PdfFileReader(file(args.image, "rb"))
    output = PdfFileWriter()
    len1 = inputpdf.getNumPages()
    lenrand = int(round(random() * len1))

    for x in range(lenrand):
        output.addPage(inputpdf.getPage(x))

    output.addPage(inputpdf2.getPage(0))

    for y in range(lenrand,len1):
        output.addPage(inputpdf.getPage(y))

    outputStream = file(args.output, "wb")
    output.write(outputStream)
    outputStream.close()

if __name__ == '__main__':
    main()