X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/dd549d9f4b46da1b7a42145f571c746f3831953e..f4c540df2eb34b761e08f67396c22daf3b14ca10:/apps/catalogue/views.py

diff --git a/apps/catalogue/views.py b/apps/catalogue/views.py
index 9168099ca..64aada533 100644
--- a/apps/catalogue/views.py
+++ b/apps/catalogue/views.py
@@ -781,12 +781,8 @@ def tag_info(request, id):
 
 def download_zip(request, format, slug):
     url = None
-    if format == 'pdf':
-        url = models.Book.zip_pdf()
-    elif format == 'epub':
-        url = models.Book.zip_epub()
-    elif format == 'mobi':
-        url = models.Book.zip_mobi()
+    if format in ('pdf', 'epub', 'mobi'):
+        url = models.Book.zip_format(format)
     elif format == 'audiobook' and slug is not None:
         book = models.Book.objects.get(slug=slug)
         url = book.zip_audiobooks()