Extract PDF - Python script to extract pages of PDF into separate files

Extract_PDF.py

import os
from PyPDF2 import PdfFileReader, PdfFileWriter

orig = input("enter path to file:\n")
pages_at_a_time = int(input('how many pages per file?\n'))

path, filename = os.path.split(orig)
path += '\\split'
filename = filename[:-4]

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

pdf_reader = PdfFileReader(orig, "rb")
num_pages = pdf_reader.getNumPages()



i = 0
pdf_writer = PdfFileWriter()

for page in range(num_pages):
    pdf_writer.addPage(pdf_reader.getPage(page))
    i += 1

    if i % pages_at_a_time == 0 or i == num_pages - 1:
        with open(f'{path}\\{filename}_{page}.pdf', 'wb') as out:
            pdf_writer.write(out)

        pdf_writer = PdfFileWriter()

print('Done')