mergePDF.py

#!/usr/bin/env python

from PyPDF2 import PdfFileReader, PdfFileWriter
import pikepdf

def remove_page(path, output, page_to_Remove):
    pdf_writer = PdfFileWriter()
    pdf_reader = PdfFileReader(path)
    idx = 1
    for page in range(pdf_reader.getNumPages()):
            # Add each page to the writer object
            if(idx!=page_to_Remove):
                print("Adding Page: "+str(idx))
                pdf_writer.addPage(pdf_reader.getPage(page))
            idx = idx+1
    
    with open(output, 'wb') as out:
        pdf_writer.write(out)


def merge_pdfs(paths, output):
    pdf_writer = PdfFileWriter()

    for path in paths:
        pdf_reader = PdfFileReader(path)        
        for page in range(pdf_reader.getNumPages()):
            # Add each page to the writer object            
            pdf_writer.addPage(pdf_reader.getPage(page))

    # Write out the merged PDF
    with open(output, 'wb') as out:
        pdf_writer.write(out)

def extract_information(pdf_path):
    with open(pdf_path, 'rb') as f:
        pdf = PdfFileReader(f)
        information = pdf.getDocumentInfo()
        number_of_pages = pdf.getNumPages()

    # txt = f"""
    # Information about {pdf_path}: 

    # Author: {information.author}
    # Creator: {information.creator}
    # Producer: {information.producer}
    # Subject: {information.subject}
    # Title: {information.title}
    # Number of pages: {number_of_pages}
    # """

    print(txt)
    return information

def decrypt_pdf(input_path, output_path, password):
  with open(input_path, 'rb') as input_file, \
    open(output_path, 'wb') as output_file:
    reader = PdfFileReader(input_file)
    reader.decrypt(password)

    writer = PdfFileWriter()

    for i in range(reader.getNumPages()):
      writer.addPage(reader.getPage(i))

    writer.write(output_file)

def decrypt_pdf2(input_path, output_path, password):
    with pikepdf.open(input_path, password) as pdf:
        num_pages = len(pdf.pages)
        print("Total pages:", num_pages)

if __name__ == '__main__':
    doc_1 = '/Users/anshul/Downloads/input01.pdf'
    doc_2 = '/Users/anshul/Downloads/input02.pdf'    
    #extract_information('/Users/anshul/Downloads/abc.pdf')
    paths = [doc_1, doc_2]
    merge_pdfs(paths, output='/Users/anshul/Downloads/output.pdf')
    #remove_page(doc_1,doc_2,1)
    #encrypted = '/Users/anshul/Downloads/abc.pdf'
    #decrypted = '/Users/anshul/Downloads/abc.pdf'
    ##password = 'password'
    #decrypt_pdf(encrypted, decrypted, password)