pdf_oxide 0.3.24

# Search for a term across all pages of a PDF and print matches.
# Run: python main.py document.pdf "query"

import sys

from pdf_oxide import PdfDocument


def main():
    if len(sys.argv) < 3:
        print("Usage: python main.py <file.pdf> <query>", file=sys.stderr)
        sys.exit(1)

    path = sys.argv[1]
    query = sys.argv[2]
    doc = PdfDocument(path)

    pages = doc.page_count
    print(f'Searching for "{query}" in {path} ({pages} pages)...\n')

    total = 0
    pages_with_hits = 0

    for i in range(pages):
        results = doc.search(query, page_index=i)
        if not results:
            continue
        pages_with_hits += 1
        print(f"Page {i + 1}: {len(results)} match(es)")
        for r in results:
            print(f'  - "...{r.context}..."')
            total += 1
        print()

    print(f"Found {total} total matches across {pages_with_hits} pages.")


if __name__ == "__main__":
    main()