X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/ea656434007c3ec64f0e583c65e177b08403c865..676af187437d5baf0543cadc200efb476b062e4b:/apps/funding/views.py

diff --git a/apps/funding/views.py b/apps/funding/views.py
index 047be49ab..bf2a073dc 100644
--- 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.generic import TemplateView, FormView, DetailView, ListView
-from getpaid.forms import PaymentMethodForm
-from .forms import DummyForm
+import getpaid.backends.payu
+from getpaid.models import Payment
+from .forms import FundingForm
 from .models import Offer, Spent, Funding
 
 
@@ -50,37 +51,39 @@ 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
 
 
 class OfferDetailView(FormView):
-    form_class = DummyForm
+    form_class = FundingForm
     template_name = "funding/offer_detail.html"
+    backend = 'getpaid.backends.payu'
 
     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):
@@ -94,12 +97,25 @@ class OfferDetailView(FormView):
         ctx['object'] = self.object
         if self.object.is_current():
             ctx['funding_no_show_current'] = True
-            ctx['payment_form'] = PaymentMethodForm('PLN', initial={'order': self.object})
         return ctx
 
     def form_valid(self, form):
         funding = form.save()
-        return redirect(funding.get_absolute_url())
+        # 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):
+    def dispatch(self, request, slug=None):
+        self.object = Offer.current()
+        if self.object is None:
+            raise Http404
+        elif slug != self.object.slug:
+            return redirect(reverse('funding_current', args=[self.object.slug]))
+        return super(CurrentView, self).dispatch(request, slug)
 
 
 class OfferListView(ListView):
@@ -111,17 +127,14 @@ class OfferListView(ListView):
         return ctx
 
 
-class FundingView(DetailView):
-    model = Funding
-
-    @method_decorator(never_cache)
-    def dispatch(self, *args, **kwargs):
-        return super(FundingView, self).dispatch(*args, **kwargs)
+class ThanksView(TemplateView):
+    template_name = "funding/thanks.html"
 
     def get_context_data(self, *args, **kwargs):
-        ctx = super(FundingView, self).get_context_data(*args, **kwargs)
-        if self.object.offer.is_current():
-            ctx['funding_no_show_current'] = True
-            ctx['payment_form'] = PaymentMethodForm('PLN', initial={'order': self.object})
+        ctx = super(ThanksView, self).get_context_data(*args, **kwargs)
+        ctx['offer'] = Offer.current()
+        ctx['funding_no_show_current'] = True
         return ctx
 
+class NoThanksView(TemplateView):
+    template_name = "funding/no_thanks.html"