minor fix

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

diff --git a/src/club/views.py b/src/club/views.py
index 74d86de..1ab1eae 100644 (file)
--- a/src/club/views.py
+++ b/src/club/views.py
@@ -1,18 +1,20 @@
-# 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 Wolne Lektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
 #
 from django.conf import settings
 from django.contrib.auth.decorators import login_required, permission_required
 from django.db.models import Sum
 from django.http import HttpResponseRedirect
 from django.shortcuts import get_object_or_404, render
+from django.urls import reverse
 from django.utils.decorators import method_decorator
 from django.views.decorators.cache import never_cache
 from django.views.generic import FormView, CreateView, TemplateView, DetailView, UpdateView
 from django.views import View
 from .payu import POSS
 from .payu import views as payu_views
-from .forms import ScheduleForm, PayUCardTokenForm
+from .forms import PayUCardTokenForm
+from . import forms
 from . import models
 from .helpers import get_active_schedule
 from .payment_methods import recurring_payment_method
@@ -27,15 +29,48 @@ class ClubView(TemplateView):
         return ctx
 
 
+
+class DonationStep1(UpdateView):
+    queryset = models.Schedule.objects.filter(payed_at=None)
+    form_class = forms.DonationStep1Form
+    slug_field = slug_url_kwarg = 'key'
+    template_name = 'club/donation_step1.html'
+    step = 1
+
+    def get_context_data(self, **kwargs):
+        c = super().get_context_data(**kwargs)
+        c['club'] = models.Club.objects.first()
+        return c
+
+    def get_success_url(self):
+        return reverse('donation_step2', args=[self.object.key])
+
+
+class DonationStep2(UpdateView):
+    queryset = models.Schedule.objects.filter(payed_at=None)
+    form_class = forms.DonationStep2Form
+    slug_field = slug_url_kwarg = 'key'
+    template_name = 'club/donation_step2.html'
+    step = 2
+
+    def get_context_data(self, **kwargs):
+        c = super().get_context_data(**kwargs)
+        c['club'] = models.Club.objects.first()
+        return c
+
+
 class JoinView(CreateView):
-    form_class = ScheduleForm
-    template_name = 'club/membership_form.html'
+    form_class = forms.DonationStep1Form
+    template_name = 'club/donation_step1.html'
+
+    @property
+    def club(self):
+        return models.Club.objects.first()
 
     def is_app(self):
         return self.request.GET.get('app')
 
     def get(self, request):
-        # TODO: configure as app-allowed hosts.
         if settings.CLUB_APP_HOST and self.is_app() and request.META['HTTP_HOST'] != settings.CLUB_APP_HOST:
             return HttpResponseRedirect("https://" + settings.CLUB_APP_HOST + request.get_full_path())
 
@@ -43,7 +78,8 @@ class JoinView(CreateView):
             request.session['from_app'] = True
         elif request.session and 'from_app' in request.session:
             del request.session['from_app']
-        return super(JoinView, self).get(request)
+
+        return super().get(request)
 
     def get_context_data(self, **kwargs):
         c = super(JoinView, self).get_context_data(**kwargs)
@@ -55,14 +91,15 @@ class JoinView(CreateView):
         return c
 
     def get_initial(self):
+        # referer?
         if self.request.user.is_authenticated and self.request.user.email:
             return {
-                'email': self.request.user.email,
+                'email': self.request.user.email
             }
 
     def get_form_kwargs(self):
         kwargs = super().get_form_kwargs()
-        kwargs['referer'] = self.request.META.get('HTTP_REFERER', '')
+        #kwargs['referer'] = self.request.META.get('HTTP_REFERER', '')
         return kwargs
 
     def form_valid(self, form):
@@ -73,15 +110,21 @@ class JoinView(CreateView):
         return retval
 
     def get_success_url(self):
-        return self.object.initiate_payment(self.request)
+        return reverse('donation_step2', args=[self.object.key])
 
 
 @method_decorator(never_cache, name='dispatch')
 class ScheduleView(DetailView):
-    model = models.Schedule
+    queryset = models.Schedule.objects.exclude(email='')
     slug_field = slug_url_kwarg = 'key'
     template_name = 'club/schedule.html'
-
+    step = 3
+    
+    def get_template_names(self):
+        if not self.object.payed_at:
+            return 'club/donation_step3.html'
+        return 'club/schedule.html'
+        
     def get_context_data(self, *args, **kwargs):
         ctx = super().get_context_data(*args, **kwargs)
         ctx['active_menu_item'] = 'club'
@@ -107,26 +150,14 @@ def cancel(request, key):
     return HttpResponseRedirect(schedule.get_absolute_url())
 
 
-class DummyPaymentView(TemplateView):
-    template_name = 'club/dummy_payment.html'
-
-    def get_context_data(self, key):
-        return {
-            'schedule': models.Schedule.objects.get(key=key),
-        }
-
-    def post(self, request, key):
-        schedule = models.Schedule.objects.get(key=key)
-        schedule.create_payment()
-        return HttpResponseRedirect(schedule.get_absolute_url())
-
-
 class PayUPayment(DetailView):
     model = models.Schedule
     slug_field = slug_url_kwarg = 'key'
 
     def get(self, request, key):
         schedule = self.get_object()
+        schedule.method = 'payu'
+        schedule.save(update_fields=['method'])
         return HttpResponseRedirect(schedule.initiate_payment(request))
 
 
@@ -142,7 +173,10 @@ class PayURecPayment(payu_views.RecPayment):
         return POSS[pos_id]
 
     def get_success_url(self):
-        return self.get_schedule().pay(self.request)
+        schedule = self.get_schedule()
+        schedule.method = 'payu-re'
+        schedule.save(update_fields=['method'])
+        return schedule.pay(self.request)
 
 
 class PayUNotifyView(payu_views.NotifyView):
@@ -151,8 +185,9 @@ class PayUNotifyView(payu_views.NotifyView):
 
 class ScheduleThanksView(DetailView):
     model = models.Schedule
+    template_name = 'club/donation_step4.html'
     slug_field = slug_url_kwarg = 'key'
-    template_name = 'club/thanks.html'
+    step = 4
 
     def get_context_data(self, *args, **kwargs):
         ctx = super().get_context_data(*args, **kwargs)