NeoAnd/Docling
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(pypdfium2): just forward input to PdfDocument directly (#17)

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-25 08:54:57 +02:00 committed by GitHub
parent 6db2b350dd
commit 3eca8b8485
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)