Django 1.9.

[wolnelektury.git] / src / funding / views.py

diff --git a/src/funding/views.py b/src/funding/views.py
index 7c9adef..74530a4 100644 (file)
--- a/src/funding/views.py
+++ b/src/funding/views.py
@@ -88,14 +88,16 @@ class OfferDetailView(FormView):
                     raise Http404
         return super(OfferDetailView, self).dispatch(request, slug)
 
-    def get_form(self, form_class):
+    def get_form(self, form_class=None):
+        if form_class is None:
+            form_class = self.get_form_class()
         if self.request.method == 'POST':
             return form_class(self.object, self.request.POST)
         else:
             return form_class(self.object, initial={'amount': app_settings.DEFAULT_AMOUNT})
 
-    def get_context_data(self, *args, **kwargs):
-        ctx = super(OfferDetailView, self).get_context_data(*args, **kwargs)
+    def get_context_data(self, **kwargs):
+        ctx = super(OfferDetailView, self).get_context_data(**kwargs)
         ctx['object'] = self.object
         ctx['page'] = self.request.GET.get('page', 1)
         if self.object.is_current():
@@ -125,8 +127,8 @@ class CurrentView(OfferDetailView):
 class OfferListView(ListView):
     queryset = Offer.public()
 
-    def get_context_data(self, *args, **kwargs):
-        ctx = super(OfferListView, self).get_context_data(*args, **kwargs)
+    def get_context_data(self, **kwargs):
+        ctx = super(OfferListView, self).get_context_data(**kwargs)
         ctx['funding_no_show_current'] = True
         return ctx
 
@@ -134,8 +136,8 @@ class OfferListView(ListView):
 class ThanksView(TemplateView):
     template_name = "funding/thanks.html"
 
-    def get_context_data(self, *args, **kwargs):
-        ctx = super(ThanksView, self).get_context_data(*args, **kwargs)
+    def get_context_data(self, **kwargs):
+        ctx = super(ThanksView, self).get_context_data(**kwargs)
         ctx['offer'] = Offer.current()
         ctx['funding_no_show_current'] = True
         return ctx
@@ -150,8 +152,7 @@ class DisableNotifications(TemplateView):
 
     @csrf_exempt
     def dispatch(self, request):
-        self.object = get_object_or_404(Funding,
-            email=request.GET.get('email'), notify_key=request.GET.get('key'))
+        self.object = get_object_or_404(Funding, email=request.GET.get('email'), notify_key=request.GET.get('key'))
         return super(DisableNotifications, self).dispatch(request)
 
     def post(self, *args, **kwargs):
@@ -159,11 +160,12 @@ class DisableNotifications(TemplateView):
         return redirect(self.request.get_full_path())
 
 
-def offer_bar(request, pk, link=False, closeable=False, show_title=True, show_title_calling=True, add_class=""):
-    offer = get_object_or_404(Offer, pk=pk)
-    offer_sum = offer.sum()
-
-    return render(request, "funding/includes/funding.html", {
+def offer_bar(offer, link=False, closeable=False, show_title=True, show_title_calling=True, add_class=""):
+    if offer:
+        offer_sum = offer.sum()
+    else:
+        return {}
+    return {
         'offer': offer,
         'sum': offer_sum,
         'is_current': offer.is_current(),
@@ -175,41 +177,40 @@ def offer_bar(request, pk, link=False, closeable=False, show_title=True, show_ti
         'show_title': show_title,
         'show_title_calling': show_title_calling,
         'add_class': add_class,
-    })
+    }
+
+
+def render_offer_bar(request, pk, link=False, closeable=False, show_title=True, show_title_calling=True, add_class=""):
+    offer = get_object_or_404(Offer, pk=pk)
+    return render(request, "funding/includes/funding.html",
+                  offer_bar(offer, link, closeable, show_title, show_title_calling, add_class))
 
 
 @ssi_included(patch_response=[ssi_cache_control(must_revalidate=True)])
 def top_bar(request, pk):
-    return offer_bar(request, pk,
-        link=True, closeable=True, add_class="funding-top-header")
+    return render_offer_bar(request, pk, link=True, closeable=True, add_class="funding-top-header")
 
 
 @ssi_included(patch_response=[ssi_cache_control(must_revalidate=True)])
 def list_bar(request, pk):
-    return offer_bar(request, pk,
-        link=True, show_title_calling=False)
+    return render_offer_bar(request, pk, link=True, show_title_calling=False)
 
 
 @ssi_included(patch_response=[ssi_cache_control(must_revalidate=True)])
 def detail_bar(request, pk):
-    return offer_bar(request, pk,
-        show_title=False)
+    return render_offer_bar(request, pk, show_title=False)
 
 
 @ssi_included(patch_response=[ssi_cache_control(must_revalidate=True)])
 def offer_status(request, pk):
     offer = get_object_or_404(Offer, pk=pk)
-    return render(request, "funding/includes/offer_status.html", {
-        'offer': offer,
-    })
+    return render(request, "funding/includes/offer_status.html", {'offer': offer})
 
 
 @ssi_included(patch_response=[ssi_cache_control(must_revalidate=True)])
 def offer_status_more(request, pk):
     offer = get_object_or_404(Offer, pk=pk)
-    return render(request, "funding/includes/offer_status_more.html", {
-        'offer': offer,
-    })
+    return render(request, "funding/includes/offer_status_more.html", {'offer': offer})
 
 
 @ssi_included(patch_response=[ssi_cache_control(must_revalidate=True)])