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

BulkDownloadView supports zipping and merging of documents

Browse Source
This commit is contained in:
Łukasz Czyż 2023-12-13 16:59:57 +01:00
parent cec58a9cbc
commit 7ddcaac4cd
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/documents/views.py
View File

@ -59,6 +59,7 @@ from rest_framework.viewsets import ViewSet
from documents import bulk_edit from documents import bulk_edit
from documents.bulk_download import ArchiveOnlyStrategy from documents.bulk_download import ArchiveOnlyStrategy
from documents.bulk_download import MergedPdfFile
from documents.bulk_download import OriginalAndArchiveStrategy from documents.bulk_download import OriginalAndArchiveStrategy
from documents.bulk_download import OriginalsOnlyStrategy from documents.bulk_download import OriginalsOnlyStrategy
from documents.classifier import load_classifier from documents.classifier import load_classifier
@ -1076,6 +1077,7 @@ class BulkDownloadView(GenericAPIView):
compression = serializer.validated_data.get("compression") compression = serializer.validated_data.get("compression")
content = serializer.validated_data.get("content") content = serializer.validated_data.get("content")
follow_filename_format = serializer.validated_data.get("follow_formatting") follow_filename_format = serializer.validated_data.get("follow_formatting")
single_file = serializer.validated_data.get("single_file")
os.makedirs(settings.SCRATCH_DIR, exist_ok=True) os.makedirs(settings.SCRATCH_DIR, exist_ok=True)
temp = tempfile.NamedTemporaryFile( temp = tempfile.NamedTemporaryFile(
@ -1091,18 +1093,21 @@ class BulkDownloadView(GenericAPIView):
else: else:
strategy_class = ArchiveOnlyStrategy strategy_class = ArchiveOnlyStrategy
with zipfile.ZipFile(temp.name, "w", compression) as zipf: def _get_bulk_document_writer():
strategy = strategy_class(zipf, follow_filename_format) if single_file:
return MergedPdfFile(temp.name)
else:
return zipfile.ZipFile(temp.name, "w", compression)
with _get_bulk_document_writer() as writer:
strategy = strategy_class(writer, follow_filename_format)
for id in ids: for id in ids:
doc = Document.objects.get(id=id) doc = Document.objects.get(id=id)
strategy.add_document(doc) strategy.add_document(doc)
with open(temp.name, "rb") as f: with open(temp.name, "rb") as f:
response = HttpResponse(f, content_type="application/zip") response = HttpResponse(f, content_type="application/zip")
response["Content-Disposition"] = '{}; filename="{}"'.format( response["Content-Disposition"] = "attachment"
"attachment",
"documents.zip",
)
return response return response