librarian update

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

diff --git a/apps/funding/views.py b/apps/funding/views.py
index 66c34a2..dbbe219 100644 (file)
--- a/apps/funding/views.py
+++ b/apps/funding/views.py
@@ -2,15 +2,16 @@
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
+from datetime import date
 from django.views.decorators.cache import never_cache
-from django.conf import settings
 from django.core.urlresolvers import reverse
 from django.http import Http404
 from django.shortcuts import redirect, get_object_or_404
-from django.utils.decorators import method_decorator
+from django.views.decorators.csrf import csrf_exempt
 from django.views.generic import TemplateView, FormView, DetailView, ListView
 import getpaid.backends.payu
 from getpaid.models import Payment
+from . import app_settings
 from .forms import FundingForm
 from .models import Offer, Spent, Funding
 
@@ -51,22 +52,22 @@ class WLFundView(TemplateView):
 
         ctx = super(WLFundView, self).get_context_data()
         offers = []
-        for o in Offer.objects.all():
-            if o.state() == 'lose':
-                o.wlfund = o.sum()
+        
+        for o in Offer.past():
+            if o.is_win():
+                o.wlfund = o.sum() - o.target
                 if o.wlfund > 0:
                     offers.append(o)
-            elif o.state() == 'win':
-                o.wlfund = o.sum() - o.target
+            else:
+                o.wlfund = o.sum()
                 if o.wlfund > 0:
                     offers.append(o)
         amount = sum(o.wlfund for o in offers) - sum(o.amount for o in Spent.objects.all())
-        print offers
 
         ctx['amount'] = amount
         ctx['log'] = add_total(amount, mix(
             (offers, lambda x: x.end, 'offer'),
-            (Spent.objects.all(), lambda x: x.timestamp, 'spent'),
+            (Spent.objects.all().select_related(), lambda x: x.timestamp, 'spent'),
         ))
         return ctx
 
@@ -76,20 +77,22 @@ class OfferDetailView(FormView):
     template_name = "funding/offer_detail.html"
     backend = 'getpaid.backends.payu'
 
+    @csrf_exempt
     def dispatch(self, request, slug=None):
-        if slug:
-            self.object = get_object_or_404(Offer.public(), slug=slug)
-        else:
-            self.object = Offer.current()
-            if self.object is None:
-                raise Http404
+        if getattr(self, 'object', None) is None:
+            if slug:
+                self.object = get_object_or_404(Offer.public(), slug=slug)
+            else:
+                self.object = Offer.current()
+                if self.object is None:
+                    raise Http404
         return super(OfferDetailView, self).dispatch(request, slug)
 
     def get_form(self, form_class):
         if self.request.method == 'POST':
             return form_class(self.object, self.request.POST)
         else:
-            return form_class(self.object, initial={'amount': settings.FUNDING_DEFAULT})
+            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)
@@ -102,9 +105,20 @@ class OfferDetailView(FormView):
         funding = form.save()
         # Skip getpaid.forms.PaymentMethodForm, go directly to the broker.
         payment = Payment.create(funding, self.backend)
-        gateway_url = payment.get_processor()(payment).get_gateway_url(self.request)
+        gateway_url_tuple = payment.get_processor()(payment).get_gateway_url(self.request)
         payment.change_status('in_progress')
-        return redirect(gateway_url)
+        return redirect(gateway_url_tuple[0])
+
+
+class CurrentView(OfferDetailView):
+    @csrf_exempt
+    def dispatch(self, request, slug=None):
+        self.object = Offer.current()
+        if self.object is None:
+            return redirect(reverse('funding'))
+        elif slug != self.object.slug:
+            return redirect(reverse('funding_current', args=[self.object.slug]))
+        return super(CurrentView, self).dispatch(request, slug)
 
 
 class OfferListView(ListView):
@@ -125,5 +139,20 @@ class ThanksView(TemplateView):
         ctx['funding_no_show_current'] = True
         return ctx
 
+
 class NoThanksView(TemplateView):
     template_name = "funding/no_thanks.html"
+
+
+class DisableNotifications(TemplateView):
+    template_name = "funding/disable_notifications.html"
+
+    @csrf_exempt
+    def dispatch(self, request):
+        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):
+        self.object.disable_notifications()
+        return redirect(self.request.get_full_path())