split_pdf_functionality.py

# importing the required modules
import PyPDF2
def split_pdf(file_path,save_path,save_name):
    def PDFsplit(pdf, splits, pdfFileObj, pdfReader):
        
        # starting index of first slice
        start = 0
        
        # starting index of last slice
        end = splits[0]
        
        
        for i in range(len(splits)):
            # creating pdf writer object for (i+1)th split
            pdfWriter = PyPDF2.PdfFileWriter()
            
            # output pdf file name
            outputpdf = save_path+"/"+save_name +"("+ str(i) +")"+ '.pdf'
            #print(outputpdf)
            # adding pages to pdf writer object
            for page in range(start,end):
                pdfWriter.addPage(pdfReader.getPage(page))
            
            # writing split pdf pages to pdf file
            with open(outputpdf, "wb") as f:
                pdfWriter.write(f)

            # interchanging page split start position for next split
            start = end
            try:
                # setting split end position for next split
                end = splits[i+1]
            except IndexError:
                # setting split end position for last split
                end = pdfReader.numPages
            
        # closing the input pdf file object
        pdfFileObj.close()
        #print("splited.")
                

    # pdf file to split
    pdf = file_path
    
    # creating input pdf file object
    pdfFileObj = open(pdf, 'rb')
    
    # creating pdf reader object
    pdfReader = PyPDF2.PdfFileReader(pdfFileObj)
    
    # split page positions///[split from # page,Total no. of pages]
    splits = [1,pdfReader.numPages]
    
    # calling PDFsplit function to split pdf
    PDFsplit(pdf, splits, pdfFileObj, pdfReader)