python - create subset PDFs for error checking

write_pages_2.py

import sys
import os
 
from PyPDF2 import PdfFileReader, PdfFileWriter
 
inpfn = sys.argv[1]
 
file_base_name = inpfn.replace('.pdf','')
pdf = PdfFileReader(inpfn)
 
pages = [0,1,2,3,4,5]
pdfWriter = PdfFileWriter()
 
for page_num in pages:
    pdfWriter.addPage(pdf.getPage(page_num))
 
with open('{0}_subset.pdf'.format(file_base_name), 'wb') as f:
    pdfWriter.write(f)
    f.close