Simple upgrades.

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

diff --git a/apps/funding/views.py b/apps/funding/views.py
index dab996a..9dd18cc 100644 (file)
--- a/apps/funding/views.py
+++ b/apps/funding/views.py
@@ -1,21 +1,31 @@
-# Create your views here.
-from django.views.generic import TemplateView
-from .models import Offer
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from django.core.urlresolvers import reverse
+from django.http import Http404
+from django.shortcuts import redirect, get_object_or_404
+from django.views.decorators.csrf import csrf_exempt
+from django.views.generic import TemplateView, FormView, ListView
+from getpaid.models import Payment
+from . import app_settings
+from .forms import FundingForm
+from .models import Offer, Spent, Funding

 
 
 def mix(*streams):
     substreams = []
 
 
 def mix(*streams):
     substreams = []

-    for stream, read_date in streams:
+    for stream, read_date, tag in streams:

         iterstream = iter(stream)
         try:
             item = next(iterstream)
         except StopIteration:
             pass
         else:
         iterstream = iter(stream)
         try:
             item = next(iterstream)
         except StopIteration:
             pass
         else:

-            substreams.append([read_date(item), item, iterstream, read_date])
+            substreams.append([read_date(item), item, iterstream, read_date, tag])

     while substreams:
     while substreams:

-        i, substream = max(enumerate(substreams), key=lambda x: x[0])
-        yield substream[1]
+        i, substream = max(enumerate(substreams), key=lambda x: x[1][0])
+        yield substream[4], substream[1]

         try:
             item = next(substream[2])
         except StopIteration:
         try:
             item = next(substream[2])
         except StopIteration:


@@ -28,12 +38,118 @@ class WLFundView(TemplateView):
     template_name = "funding/wlfund.html"
 
     def get_context_data(self):
     template_name = "funding/wlfund.html"
 
     def get_context_data(self):

+        def add_total(total, it):
+            for tag, e in it:
+                e.total = total
+                if tag == 'spent':
+                    total += e.amount
+                else:
+                    total -= e.wlfund
+                yield tag, e
+

         ctx = super(WLFundView, self).get_context_data()
         ctx = super(WLFundView, self).get_context_data()

-        offers = [o for o in Offer.objects.all() if o.state() == 'lose' and o.sum()]
-        amount = sum(o.sum() for o in offers)
-        print offers
+        offers = []
+
+        for o in Offer.past():
+            if o.is_win():
+                o.wlfund = o.sum() - o.target
+                if o.wlfund > 0:
+                    offers.append(o)
+            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())

 
 

-        #offers = (o for o in Offer.objects.all() if o.state() == 'lose' and o.sum())

         ctx['amount'] = amount
         ctx['amount'] = amount

-        ctx['log'] = mix((offers, lambda x: x.end))
+        ctx['log'] = add_total(amount, mix(
+            (offers, lambda x: x.end, 'offer'),
+            (Spent.objects.all().select_related(), lambda x: x.timestamp, 'spent'),
+        ))
+        return ctx
+
+
+class OfferDetailView(FormView):
+    form_class = FundingForm
+    template_name = "funding/offer_detail.html"
+    backend = 'getpaid.backends.payu'
+
+    @csrf_exempt
+    def dispatch(self, request, slug=None):
+        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': app_settings.DEFAULT_AMOUNT})
+
+    def get_context_data(self, *args, **kwargs):
+        ctx = super(OfferDetailView, self).get_context_data(*args, **kwargs)
+        ctx['object'] = self.object
+        if self.object.is_current():
+            ctx['funding_no_show_current'] = True
+        return ctx
+
+    def form_valid(self, form):
+        funding = form.save()
+        # Skip getpaid.forms.PaymentMethodForm, go directly to the broker.
+        payment = Payment.create(funding, self.backend)
+        gateway_url_tuple = payment.get_processor()(payment).get_gateway_url(self.request)
+        payment.change_status('in_progress')
+        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):
+    queryset = Offer.public()
+
+    def get_context_data(self, *args, **kwargs):
+        ctx = super(OfferListView, self).get_context_data(*args, **kwargs)
+        ctx['funding_no_show_current'] = True
+        return ctx
+
+
+class ThanksView(TemplateView):
+    template_name = "funding/thanks.html"
+
+    def get_context_data(self, *args, **kwargs):
+        ctx = super(ThanksView, self).get_context_data(*args, **kwargs)
+        ctx['offer'] = Offer.current()
+        ctx['funding_no_show_current'] = True

         return ctx
         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())