File size: 1,687 Bytes
5dccc28 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | from pathlib import Path
from io import BytesIO
from pypdf import PdfReader
def extract_pdf_pages(file_path: Path, max_pages: int | None = None) -> tuple[list[tuple[int, str]], int]:
reader = PdfReader(str(file_path))
total_pages = len(reader.pages)
page_limit = min(total_pages, max_pages) if max_pages else total_pages
pages: list[tuple[int, str]] = []
for index, page in enumerate(reader.pages[:page_limit], start=1):
pages.append((index, (page.extract_text() or "").strip()))
return pages, total_pages
def extract_pdf_text(file_path: Path, max_pages: int | None = None) -> tuple[str, int]:
pages, total_pages = extract_pdf_pages(file_path, max_pages=max_pages)
text = "\n\n".join(page_text for _, page_text in pages).strip()
return text, total_pages
def extract_pdf_pages_from_bytes(content: bytes, max_pages: int | None = None) -> tuple[list[tuple[int, str]], int]:
reader = PdfReader(BytesIO(content))
total_pages = len(reader.pages)
page_limit = min(total_pages, max_pages) if max_pages else total_pages
pages: list[tuple[int, str]] = []
for index, page in enumerate(reader.pages[:page_limit], start=1):
pages.append((index, (page.extract_text() or "").strip()))
return pages, total_pages
def extract_pdf_text_from_bytes(content: bytes, max_pages: int | None = None) -> tuple[str, int]:
pages, total_pages = extract_pdf_pages_from_bytes(content, max_pages=max_pages)
text = "\n\n".join(page_text for _, page_text in pages).strip()
return text, total_pages
def count_pdf_pages_from_bytes(content: bytes) -> int:
reader = PdfReader(BytesIO(content))
return len(reader.pages)
|