From 8ae7c36dcdaa17301f8085b69b003d513ffcfa1f Mon Sep 17 00:00:00 2001 From: =?utf8?q?Marek=20St=C4=99pniowski?= Date: Thu, 18 Sep 2008 15:34:50 +0200 Subject: [PATCH] Now shelf archive contains all pdf, odt and txt formats from books. --- apps/catalogue/views.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/apps/catalogue/views.py b/apps/catalogue/views.py index de75803a4..0026c173b 100644 --- a/apps/catalogue/views.py +++ b/apps/catalogue/views.py @@ -214,13 +214,21 @@ def download_shelf(request, slug): temp = tempfile.TemporaryFile() archive = zipfile.ZipFile(temp, 'w', zipfile.ZIP_DEFLATED) for book in models.Book.tagged.with_all(shelf): - filename = book.html_file.path - archive.write(filename, str('%s.html' % book.slug)) + if book.pdf_file: + filename = book.pdf_file.path + archive.write(filename, str('%s.pdf' % book.slug)) + if book.odt_file: + filename = book.odt_file.path + archive.write(filename, str('%s.odt' % book.slug)) + if book.txt_file: + filename = book.txt_file.path + archive.write(filename, str('%s.txt' % book.slug)) archive.close() + # Write file to archive in small chunks wrapper = FileWrapper(temp) response = HttpResponse(wrapper, content_type='application/zip') - response['Content-Disposition'] = 'attachment; filename=%s.zip' % shelf.slug + response['Content-Disposition'] = 'attachment; filename=%s.zip' % shelf.sort_key response['Content-Length'] = temp.tell() temp.seek(0) return response -- 2.20.1