limit pdf files to specific pages

extractPages.py

from PyPDF2 import PdfFileReader, PdfFileWriter
import os

inputPath = "./solutions"
outputPath = "./extractedPages"
chosenPages = [1]

if not os.path.exists(outputPath):
    os.makedirs(outputPath)

for fileName in os.listdir(inputPath):
    pdf = PdfFileReader(inputPath + "/" + fileName)
    pdfWriter = PdfFileWriter()
    for page_num in chosenPages:
        pdfWriter.addPage(pdf.getPage(page_num))
    with open(outputPath + "/" + fileName, "wb") as f:
        pdfWriter.write(f)

print("Extracted the pages", chosenPages, "from the pdfs")