PDF page selection

import PyPDF2

def pdf_page_select(input_pdf, select_pages):
    # open source pdf file
    pdfFileObj = open(input_pdf, 'rb')

    # creating pdf reader object for page selection
    pdfReader = PyPDF2.PdfFileReader(pdfFileObj)

    # creating pdf writer object to write new pdf file
    pdfWriter = PyPDF2.PdfFileWriter()

    # output pdf file name
    outputpdf = 'Ready_to_print.pdf'

    # adding pages to pdf writer object
    for page_index in select_pages:
        pdfWriter.addPage(pdfReader.getPage(page_index-1))

    # writing split pdf pages to pdf file
    with open(outputpdf, "wb") as f:
        pdfWriter.write(f)

    # closing the input pdf file object
    pdfFileObj.close()



def main():
    # input pdf file
    pdf = 'demo_2.pdf'

    # index of wanted page
    wanted_page = [1,3,4]

    # select wanted page and output to another pdf file
    pdf_page_select(pdf, wanted_page)

if __name__ == "__main__":
    main()