lgcosta/paperless-ngx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use pikepdf as context manager

Browse Source
This commit is contained in:
shamoon 2024-09-25 00:25:25 -07:00
parent e5836118b6
commit e7de2a3d79
2 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/documents/migrations/1053_document_pages_count.py
View File

@ -31,11 +31,10 @@ def add_number_of_pages_to_pages_count(apps, schema_editor):
) )
try: try:
pdf = pikepdf.open(source_path(doc)) with pikepdf.Pdf.open(source_path(doc)) as pdf:
if pdf.pages is not None:
if pdf.pages is not None: doc.pages_count = len(pdf.pages)
doc.pages_count = len(pdf.pages) doc.save()
doc.save()
except Exception as e: # pragma: no cover except Exception as e: # pragma: no cover
print(f"Error retrieving number of pages for {doc.filename}: {e}") print(f"Error retrieving number of pages for {doc.filename}: {e}")

4
src/paperless_tesseract/parsers.py
View File

@ -46,8 +46,8 @@ class RasterisedDocumentParser(DocumentParser):
if mime_type == "application/pdf": if mime_type == "application/pdf":
import pikepdf import pikepdf
pdf = pikepdf.open(document_path) with pikepdf.Pdf.open(document_path) as pdf:
pages_count = len(pdf.pages) pages_count = len(pdf.pages)
return pages_count return pages_count
def extract_metadata(self, document_path, mime_type): def extract_metadata(self, document_path, mime_type):