107 lines
3.8 KiB
Python
107 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals, absolute_import
|
|
|
|
from django.conf import settings
|
|
from django.shortcuts import redirect
|
|
from django.views.generic import UpdateView, RedirectView
|
|
|
|
from django.contrib import messages
|
|
from django.contrib.auth.models import Group
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
|
|
from mhackspace.users.models import User, Membership
|
|
from mhackspace.users.models import MEMBERSHIP_CANCELLED
|
|
from mhackspace.users.forms import MembershipJoinForm
|
|
from mhackspace.subscriptions.payments import select_provider
|
|
|
|
|
|
class MembershipCancelView(LoginRequiredMixin, RedirectView):
|
|
permanent = False
|
|
pattern_name = 'users:detail'
|
|
|
|
def get_redirect_url(self, *args, **kwargs):
|
|
payment_provider = 'gocardless'
|
|
provider = select_provider(payment_provider)
|
|
|
|
member = Membership.objects.filter(user=self.request.user).first()
|
|
|
|
result = provider.cancel_subscription(
|
|
reference=member.reference
|
|
)
|
|
if result.get('success') is True:
|
|
# set membership to cancelled on success
|
|
member.status = MEMBERSHIP_CANCELLED
|
|
member.save()
|
|
|
|
# remove user from group on success
|
|
group = Group.objects.get(name='members')
|
|
self.request.user.groups.remove(group)
|
|
messages.add_message(
|
|
self.request,
|
|
messages.SUCCESS,
|
|
'Your membership has now been cancelled')
|
|
|
|
kwargs['username'] = self.request.user.get_username()
|
|
return super(MembershipCancelView, self).get_redirect_url(*args, **kwargs)
|
|
|
|
|
|
class MembershipJoinView(LoginRequiredMixin, UpdateView):
|
|
model = User
|
|
fields = []
|
|
|
|
def get_object(self):
|
|
# Only get the User record for the user making the request
|
|
return User.objects.get(username=self.request.user.username)
|
|
|
|
def form_valid(self, form):
|
|
app_domain = 'http://test.maidstone-hackspace.org.uk'
|
|
payment_provider = 'gocardless'
|
|
provider = select_provider(payment_provider)
|
|
user_code = str(self.request.user.id).zfill(5)
|
|
|
|
form_subscription = MembershipJoinForm(data=self.request.POST)
|
|
form_subscription.is_valid()
|
|
|
|
success_url = '%s/membership/%s/success' % (app_domain, payment_provider)
|
|
failure_url = '%s/membership/%s/failure' % (app_domain, payment_provider)
|
|
url = provider.create_subscription(
|
|
amount=form_subscription.cleaned_data.get('amount', 20.00),
|
|
name="Membership your membership id is MH%s" % user_code,
|
|
redirect_success=success_url,
|
|
redirect_failure=failure_url
|
|
)
|
|
|
|
return redirect(url)
|
|
|
|
|
|
class MembershipJoinSuccessView(LoginRequiredMixin, RedirectView):
|
|
permanent = False
|
|
pattern_name = 'users:detail'
|
|
|
|
def get_redirect_url(self, *args, **kwargs):
|
|
del(kwargs['provider'])
|
|
messages.add_message(
|
|
self.request,
|
|
messages.SUCCESS,
|
|
'Success your membership should now be active')
|
|
kwargs['username'] = self.request.user.get_username()
|
|
|
|
# add user to group on success
|
|
group = Group.objects.get(name='members')
|
|
self.request.user.groups.add(group)
|
|
return super(MembershipJoinSuccessView, self).get_redirect_url(*args, **kwargs)
|
|
|
|
|
|
class MembershipJoinFailureView(LoginRequiredMixin, RedirectView):
|
|
permanent = False
|
|
pattern_name = 'users:detail'
|
|
|
|
def get_redirect_url(self, *args, **kwargs):
|
|
del(kwargs['provider'])
|
|
messages.add_message(
|
|
self.request,
|
|
messages.ERROR,
|
|
'Failed to sign up something went wrong with your payment, please contact us at %s' % EMAIL_SUPPORT)
|
|
kwargs['username'] = self.request.user.get_username()
|
|
return super(MembershipJoinFailureView, self).get_redirect_url(*args, **kwargs)
|