migrations for django-getpaid

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

diff --git a/apps/funding/views.py b/apps/funding/views.py
index f543162..047be49 100644 (file)
--- a/apps/funding/views.py
+++ b/apps/funding/views.py
@@ -2,12 +2,16 @@
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #

+from django.views.decorators.cache import never_cache

 from django.conf import settings
 from django.core.urlresolvers import reverse
 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.shortcuts import redirect, get_object_or_404

-from django.views.generic import TemplateView, FormView, DetailView
+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
 from .forms import DummyForm

-from .models import Offer, Spent
+from .models import Offer, Spent, Funding

 
 
 def mix(*streams):
 
 
 def mix(*streams):


@@ -21,7 +25,7 @@ def mix(*streams):
         else:
             substreams.append([read_date(item), item, iterstream, read_date, tag])
     while substreams:
         else:
             substreams.append([read_date(item), item, iterstream, read_date, tag])
     while substreams:

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

         yield substream[4], substream[1]
         try:
             item = next(substream[2])
         yield substream[4], substream[1]
         try:
             item = next(substream[2])


@@ -41,12 +45,21 @@ class WLFundView(TemplateView):
                 if tag == 'spent':
                     total += e.amount
                 else:
                 if tag == 'spent':
                     total += e.amount
                 else:

-                    total -= e.sum()
+                    total -= e.wlfund

                 yield tag, e
 
         ctx = super(WLFundView, self).get_context_data()
                 yield tag, e
 
         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) - sum(o.amount for o in Spent.objects.all())
+        offers = []
+        for o in Offer.objects.all():
+            if o.state() == 'lose':
+                o.wlfund = o.sum()
+                if o.wlfund > 0:
+                    offers.append(o)
+            elif o.state() == 'win':
+                o.wlfund = o.sum() - o.target
+                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
         print offers
 
         ctx['amount'] = amount


@@ -61,8 +74,13 @@ class OfferDetailView(FormView):
     form_class = DummyForm
     template_name = "funding/offer_detail.html"
 
     form_class = DummyForm
     template_name = "funding/offer_detail.html"
 

-    def dispatch(self, request, slug):
-        self.object = get_object_or_404(Offer.public(), slug=slug)
+    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

         return super(OfferDetailView, self).dispatch(request, slug)
 
     def get_form(self, form_class):
         return super(OfferDetailView, self).dispatch(request, slug)
 
     def get_form(self, form_class):


@@ -74,17 +92,36 @@ class OfferDetailView(FormView):
     def get_context_data(self, *args, **kwargs):
         ctx = super(OfferDetailView, self).get_context_data(*args, **kwargs)
         ctx['object'] = self.object
     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
+            ctx['payment_form'] = PaymentMethodForm('PLN', initial={'order': self.object})

         return ctx
 
     def form_valid(self, form):
         return ctx
 
     def form_valid(self, form):

-        form.save()
-        return redirect(reverse("funding_thanks"))
+        funding = form.save()
+        return redirect(funding.get_absolute_url())
+
+
+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"
+class FundingView(DetailView):
+    model = Funding
+
+    @method_decorator(never_cache)
+    def dispatch(self, *args, **kwargs):
+        return super(FundingView, self).dispatch(*args, **kwargs)

 
     def get_context_data(self, *args, **kwargs):
 
     def get_context_data(self, *args, **kwargs):

-        ctx = super(ThanksView, self).get_context_data(*args, **kwargs)
-        ctx['object'] = Offer.current()
+        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})

         return ctx
         return ctx

+