src/club/views.py

   1 from django.contrib.auth.decorators import login_required
   2 from django.http import HttpResponseRedirect
   3 from django.shortcuts import get_object_or_404, render
   4 from django.views.generic import FormView, CreateView, TemplateView, DetailView, UpdateView
   5 from django.views import View
   6 from .payu import POSS
   7 from .payu import views as payu_views
   8 from .forms import ScheduleForm, PayUCardTokenForm
   9 from . import models
  10 from .helpers import get_active_schedule
  11 from .payment_methods import payure_method
  12
  13
  14 class ClubView(TemplateView):
  15     template_name = 'club/index.html'
  16
  17
  18 class JoinView(CreateView):
  19     form_class = ScheduleForm
  20     template_name = 'club/membership_form.html'
  21
  22     def is_app(self):
  23         return self.request.GET.get('app')
  24
  25     def get(self, request):
  26         if self.is_app():
  27             request.session['from_app'] = True
  28         elif request.session and 'from_app' in request.session:
  29             del request.session['from_app']
  30         schedule = get_active_schedule(request.user)
  31         if schedule is not None:
  32             return HttpResponseRedirect(schedule.get_absolute_url())
  33         else:
  34             return super(JoinView, self).get(request)
  35
  36     def get_form_kwargs(self):
  37         kwargs = super().get_form_kwargs()
  38         kwargs['request'] = self.request
  39         return kwargs
  40
  41     def get_context_data(self, **kwargs):
  42         c = super(JoinView, self).get_context_data(**kwargs)
  43         c['membership'] = getattr(self.request.user, 'membership', None)
  44         #if hasattr(form, 'errors'):
  45         #    print(form.errors)
  46         return c
  47
  48     def get_initial(self):
  49         if self.request.user.is_authenticated and self.request.user.email:
  50             return {
  51                 'email': self.request.user.email,
  52             }
  53
  54     def form_valid(self, form):
  55         retval = super(JoinView, self).form_valid(form)
  56         if self.request.user.is_authenticated:
  57             form.instance.membership, created = models.Membership.objects.get_or_create(user=self.request.user)
  58             form.instance.save()
  59         return retval
  60
  61     def get_success_url(self):
  62         return self.object.initiate_payment(self.request)
  63
  64
  65 class ScheduleView(DetailView):
  66     model = models.Schedule
  67     slug_field = slug_url_kwarg = 'key'
  68     template_name = 'club/schedule.html'
  69
  70     def post(self, request, key):
  71         schedule = self.get_object()
  72         return HttpResponseRedirect(schedule.initiate_payment(request))
  73
  74
  75 @login_required
  76 def claim(request, key):
  77     schedule = models.Schedule.objects.get(key=key, membership=None)
  78     schedule.membership, created = models.Membership.objects.get_or_create(user=request.user)
  79     schedule.save()
  80     return HttpResponseRedirect(schedule.get_absolute_url())
  81
  82
  83 def cancel(request, key):
  84     schedule = models.Schedule.objects.get(key=key)
  85     schedule.is_cancelled = True
  86     schedule.save()
  87     return HttpResponseRedirect(schedule.get_absolute_url())
  88
  89
  90 class DummyPaymentView(TemplateView):
  91     template_name = 'club/dummy_payment.html'
  92
  93     def get_context_data(self, key):
  94         return {
  95             'schedule': models.Schedule.objects.get(key=key),
  96         }
  97
  98     def post(self, request, key):
  99         schedule = models.Schedule.objects.get(key=key)
 100         schedule.create_payment()
 101         return HttpResponseRedirect(schedule.get_absolute_url())
 102
 103
 104 class PayUPayment(payu_views.Payment):
 105     pass
 106
 107
 108 class PayURecPayment(payu_views.RecPayment):
 109     form_class = PayUCardTokenForm
 110
 111     def get_schedule(self):
 112         return get_object_or_404(models.Schedule, key=self.kwargs['key'])
 113
 114     def get_pos(self):
 115         pos_id = payure_method.pos_id
 116         return POSS[pos_id]
 117
 118     def get_success_url(self):
 119         return self.get_schedule().pay(self.request)
 120
 121
 122 class PayUNotifyView(payu_views.NotifyView):
 123     order_model = models.PayUOrder
 124
 125
 126 class MembershipView(UpdateView):
 127     fields = ['name']
 128
 129     def get_success_url(self):
 130         # TODO: get only current schedule if multiple.
 131         return self.object.schedule_set.first().get_absolute_url()
 132
 133     def get_object(self):
 134         return self.request.user.membership