fix label failing to render for non-pdfs

This commit is contained in:
Tomáš Mládek 2020-04-01 20:45:51 +02:00
parent 112a1728d5
commit 24d3f4686a

View file

@ -117,7 +117,7 @@ class DocumentWithLabel(BasePileViewMixin):
final_stream = None final_stream = None
concat_succeeded = False concat_succeeded = False
if document.file: if document.is_local_pdf:
try: try:
with open(document.file.path, 'rb') as document_fp: with open(document.file.path, 'rb') as document_fp:
writer = PdfFileWriter() writer = PdfFileWriter()
@ -141,6 +141,7 @@ class DocumentWithLabel(BasePileViewMixin):
else: else:
return redirect(document.external_url) return redirect(document.external_url)
else: else:
label_stream.seek(0)
final_stream = label_stream final_stream = label_stream
return FileResponse(final_stream, return FileResponse(final_stream,