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

diff --git a/apps/catalogue/views.py b/apps/catalogue/views.py
index 168645009..6405a49a9 100644
--- a/apps/catalogue/views.py
+++ b/apps/catalogue/views.py
@@ -558,10 +558,15 @@ class CustomPDFFormView(AjaxableFormView):
         from copy import copy
         if request.method == 'POST':
             request.GET = copy(request.GET)
-            request.GET['next'] = "%s?%s" % (reverse('catalogue.views.download_custom_pdf', args=[request.GET['slug']]),
+            request.GET['next'] = "%s?%s" % (reverse('catalogue.views.download_custom_pdf', args=[request.GET.get('slug')]),
                                              request.POST.urlencode())
         return super(CustomPDFFormView, self).__call__(request)
 
+    def get_object(self, request):
+        return get_object_or_404(models.Book, slug=request.GET.get('slug'))
+
+    def context_description(self, request, obj):
+        return obj.pretty_title()
 
     def success(self, *args):
         pass