pdfrange

#!/usr/bin/env python3

import sys
from PyPDF2 import PdfReader, PdfWriter

def main():
    if len(sys.argv) != 5:
        print("Usage: pdfrange output.pdf input.pdf start end")
        sys.exit(1)

    output_pdf = sys.argv[1]
    input_pdf = sys.argv[2]
    start = int(sys.argv[3])
    end = int(sys.argv[4])

    if start < 1 or end < start:
        print("Invalid range. Make sure start >= 1 and end >= start.")
        sys.exit(1)

    try:
        reader = PdfReader(input_pdf)
        writer = PdfWriter()

        num_pages = len(reader.pages)

        if end > num_pages:
            print(f"The input PDF only has {num_pages} pages.")
            sys.exit(1)

        for i in range(start - 1, end):
            writer.add_page(reader.pages[i])

        with open(output_pdf, 'wb') as f:
            writer.write(f)

        print(f"PDF created successfully: {output_pdf}")

    except Exception as e:
        print(f"Error processing PDF: {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()