【Python学习笔记】merge_pdfs.py

Can you help me write a python program to merge pdfs?

pip install PyPDF2

import PyPDF2
from PyPDF2 import PdfMerger

def merge_pdfs(pdf_list, output_path):
    merger = PdfMerger()
    
    for pdf in pdf_list:
        merger.append(pdf)
    
    merger.write(output_path)
    merger.close()
    print(f"PDFs merged successfully into {output_path}")

if __name__ == "__main__":
    pdf_files = ["file1.pdf", "file2.pdf", "file3.pdf"]  # List your PDF files here
    output_file = "merged_output.pdf"
    merge_pdfs(pdf_files, output_file)

Can you modify the program to take the filenames as parameters?

Sure, here's the modified program that takes the filenames as command-line parameters:

import PyPDF2
from PyPDF2 import PdfMerger
import sys

def merge_pdfs(pdf_list, output_path):
    merger = PdfMerger()
    
    for pdf in pdf_list:
        merger.append(pdf)
    
    merger.write(output_path)
    merger.close()
    print(f"PDFs merged successfully into {output_path}")

if __name__ == "__main__":
    if len(sys.argv) < 3:
        print("Usage: python merge_pdfs.py output_file.pdf input_file1.pdf input_file2.pdf ...")
        sys.exit(1)
    
    output_file = sys.argv[1]
    pdf_files = sys.argv[2:]
    
    merge_pdfs(pdf_files, output_file)

python merge_pdfs.py merged_output.pdf file1.pdf file2.pdf file3.pdf

"Answer Generated by OpenAI's ChatGPT"