mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-26 20:14:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

pypdfium2: just forward input to PdfDocument directly

Browse Source 
PdfDocument() should do accept strings, paths, bytes and byte streams. If not, please file a bug report.

Signed-off-by: mara004 <geisserml@gmail.com>
This commit is contained in:
mara004 2024-07-23 16:42:02 +02:00 committed by GitHub
parent 9f2add112f
commit 2b86acec5d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
docling/backend/pypdfium2_backend.py
View File

@ -201,13 +201,7 @@ class PyPdfiumPageBackend(PdfPageBackend):
class PyPdfiumDocumentBackend(PdfDocumentBackend): class PyPdfiumDocumentBackend(PdfDocumentBackend):
def __init__(self, path_or_stream: Iterable[Union[BytesIO, Path]]): def __init__(self, path_or_stream: Iterable[Union[BytesIO, Path]]):
super().__init__(path_or_stream) super().__init__(path_or_stream)
self._pdoc = pdfium.PdfDocument(path_or_stream)
if isinstance(path_or_stream, Path):
self._pdoc = pdfium.PdfDocument(path_or_stream)
elif isinstance(path_or_stream, BytesIO):
self._pdoc = pdfium.PdfDocument(
path_or_stream
) # TODO Fix me, won't accept bytes.
def page_count(self) -> int: def page_count(self) -> int:
return len(self._pdoc) return len(self._pdoc)