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