Temporary search problem workaround.

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

diff --git a/src/club/views.py b/src/club/views.py
index af59b1e..e80acaa 100644 (file)
--- a/src/club/views.py
+++ b/src/club/views.py
@@ -1,7 +1,10 @@
+from django.conf import settings

 from django.contrib.auth.decorators import login_required
 from django.http import HttpResponseRedirect
 from django.shortcuts import get_object_or_404, render
 from django.contrib.auth.decorators import login_required
 from django.http import HttpResponseRedirect
 from django.shortcuts import get_object_or_404, render

-from django.views.generic import FormView, CreateView, TemplateView, DetailView
+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 django.views import View
 from .payu import POSS
 from .payu import views as payu_views


@@ -14,6 +17,11 @@ from .payment_methods import payure_method
 class ClubView(TemplateView):
     template_name = 'club/index.html'
 
 class ClubView(TemplateView):
     template_name = 'club/index.html'
 

+    def get_context_data(self, *args, **kwargs):
+        ctx = super().get_context_data(*args, **kwargs)
+        ctx['active_menu_item'] = 'club'
+        return ctx
+

 
 class JoinView(CreateView):
     form_class = ScheduleForm
 
 class JoinView(CreateView):
     form_class = ScheduleForm


@@ -23,6 +31,10 @@ class JoinView(CreateView):
         return self.request.GET.get('app')
 
     def get(self, request):
         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())
+

         if self.is_app():
             request.session['from_app'] = True
         elif request.session and 'from_app' in request.session:
         if self.is_app():
             request.session['from_app'] = True
         elif request.session and 'from_app' in request.session:


@@ -38,9 +50,10 @@ class JoinView(CreateView):
         kwargs['request'] = self.request
         return kwargs
 
         kwargs['request'] = self.request
         return kwargs
 

-    def get_context_data(self, form=None):
-        c = super(JoinView, self).get_context_data()
+    def get_context_data(self, **kwargs):
+        c = super(JoinView, self).get_context_data(**kwargs)

         c['membership'] = getattr(self.request.user, 'membership', None)
         c['membership'] = getattr(self.request.user, 'membership', None)

+        c['active_menu_item'] = 'club'

         return c
 
     def get_initial(self):
         return c
 
     def get_initial(self):


@@ -60,17 +73,20 @@ class JoinView(CreateView):
         return self.object.initiate_payment(self.request)
 
 
         return self.object.initiate_payment(self.request)
 
 

+@method_decorator(never_cache, name='dispatch')

 class ScheduleView(DetailView):
     model = models.Schedule
     slug_field = slug_url_kwarg = 'key'
     template_name = 'club/schedule.html'
 
 class ScheduleView(DetailView):
     model = models.Schedule
     slug_field = slug_url_kwarg = 'key'
     template_name = 'club/schedule.html'
 

+    def get_context_data(self, *args, **kwargs):
+        ctx = super().get_context_data(*args, **kwargs)
+        ctx['active_menu_item'] = 'club'
+        return ctx
+

     def post(self, request, key):
         schedule = self.get_object()
     def post(self, request, key):
         schedule = self.get_object()

-        if not schedule.is_active():
-            return HttpResponseRedirect(schedule.initiate_payment(request))
-        else:
-            return HttpResponseRedirect(schedule.get_absolute_url())
+        return HttpResponseRedirect(schedule.initiate_payment(request))

 
 
 @login_required
 
 
 @login_required


@@ -123,3 +139,25 @@ class PayURecPayment(payu_views.RecPayment):
 class PayUNotifyView(payu_views.NotifyView):
     order_model = models.PayUOrder
 
 class PayUNotifyView(payu_views.NotifyView):
     order_model = models.PayUOrder
 

+
+class MembershipView(UpdateView):
+    fields = ['name']
+
+    def get_success_url(self):
+        # TODO: get only current schedule if multiple.
+        return self.object.schedule_set.first().get_absolute_url()
+
+    def get_object(self):
+        return self.request.user.membership
+
+
+class ScheduleThanksView(DetailView):
+    model = models.Schedule
+    slug_field = slug_url_kwarg = 'key'
+    template_name = 'club/thanks.html'
+
+    def get_context_data(self, *args, **kwargs):
+        ctx = super().get_context_data(*args, **kwargs)
+        ctx['active_menu_item'] = 'club'
+        return ctx
+