from datetime import timedelta from django.shortcuts import render, redirect from django.urls import reverse from django.conf import settings from django.contrib.auth.decorators import login_required from django.http import HttpResponse, HttpResponseBadRequest, HttpResponseNotFound from django.views.decorators.csrf import csrf_exempt from django.utils import timezone from django.contrib import messages from django.utils.translation import ugettext_lazy as _ from .forms import NewPaymentForm from .models import Payment, Subscription, BACKENDS, ACTIVE_BACKENDS @login_required def new(request): if request.method != 'POST': return redirect('account:index') form = NewPaymentForm(request.POST) if not form.is_valid(): return redirect('account:index') if request.user.vpnuser.get_subscription() is not None: return redirect('account:index') subscr = form.cleaned_data['subscr'] == '1' backend_id = form.cleaned_data['method'] months = int(form.cleaned_data['time']) if backend_id not in ACTIVE_BACKENDS: return HttpResponseNotFound() if subscr: if months not in (3, 6, 12): return redirect('account:index') rps = Subscription( user=request.user, backend_id=backend_id, period=str(months) + 'm', ) rps.save() r = rps.backend.new_subscription(rps) else: payment = Payment.create_payment(backend_id, request.user, months) payment.save() r = payment.backend.new_payment(payment) if not r: payment.status = 'error' payment.save() raise Exception("Failed to initialize payment #%d" % payment.id) if isinstance(r, str): return render(request, 'payments/form.html', dict(html=r)) elif r is None: return redirect('payments:view', payment.id) return r @csrf_exempt def callback_paypal(request, id): """ PayPal IPN """ if not BACKENDS['paypal'].backend_enabled: return HttpResponseNotFound() p = Payment.objects.get(id=id) if BACKENDS['paypal'].callback(p, request): return HttpResponse() else: return HttpResponseBadRequest() @csrf_exempt @login_required def callback_stripe(request, id): """ Stripe button POST """ if not BACKENDS['stripe'].backend_enabled: return HttpResponseNotFound() p = Payment.objects.get(id=id) BACKENDS['stripe'].callback(p, request) return redirect(reverse('payments:view', args=(id,))) @csrf_exempt def callback_coinbase(request): if not BACKENDS['coinbase'].backend_enabled: return HttpResponseNotFound() if BACKENDS['coinbase'].callback(Payment, request): return HttpResponse() else: return HttpResponseBadRequest() @csrf_exempt def callback_paypal_subscr(request, id): """ PayPal Subscription IPN """ if not BACKENDS['paypal'].backend_enabled: return HttpResponseNotFound() p = Subscription.objects.get(id=id) if BACKENDS['paypal'].callback_subscr(p, request): return HttpResponse() else: return HttpResponseBadRequest() @csrf_exempt @login_required def callback_stripe_subscr(request, id): """ Stripe subscription form target """ if not BACKENDS['stripe'].backend_enabled: return HttpResponseNotFound() p = Subscription.objects.get(id=id) BACKENDS['stripe'].callback_subscr(p, request) if p.status == 'error' or p.status == 'cancelled': messages.add_message(request, messages.ERROR, _("Error subscribing. It usually means you don't" " have enough money available.")) else: messages.add_message(request, messages.INFO, _("Subscribed!")) return redirect(reverse('account:index')) @csrf_exempt def stripe_hook(request): if not BACKENDS['stripe'].backend_enabled: return HttpResponseNotFound() if BACKENDS['stripe'].webhook(request): return HttpResponse() else: return HttpResponseBadRequest() @login_required @csrf_exempt def view(request, id): p = Payment.objects.get(id=id, user=request.user) return render(request, 'payments/view.html', dict(payment=p)) @login_required def cancel(request, id): p = Payment.objects.get(id=id, user=request.user) if p.status == 'new': p.status = 'cancelled' p.save() return render(request, 'payments/view.html', dict(payment=p)) @login_required def cancel_subscr(request, id): if request.method != 'POST': return redirect('account:index') p = Subscription.objects.get(id=id, user=request.user) try: p.backend.cancel_subscription(p) messages.add_message(request, messages.INFO, _("Subscription cancelled!")) except NotImplementedError: pass return redirect('account:index') @login_required def return_subscr(request, id): p = Subscription.objects.get(id=id, user=request.user) if p.status == 'new': p.status = 'unconfirmed' p.save() return redirect('account:index') @login_required def list_payments(request): # Only show recent cancelled payments cancelled_limit = timezone.now() - timedelta(days=3) objects = request.user.payment_set.exclude(status='cancelled', created__lte=cancelled_limit) return render(request, 'payments/list.html', dict(payments=objects))