form for #cojaczytam

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

diff --git a/src/paypal/views.py b/src/paypal/views.py
index a4c04ce..6708f2d 100644 (file)
--- a/src/paypal/views.py
+++ b/src/paypal/views.py
@@ -6,22 +6,24 @@ from decimal import Decimal
 
 from django.contrib.auth.decorators import login_required
 from django.http import Http404
 
 from django.contrib.auth.decorators import login_required
 from django.http import Http404

-from django.http.response import HttpResponseRedirect
+from django.http.response import HttpResponseRedirect, HttpResponseForbidden

 from django.shortcuts import render
 
 from django.shortcuts import render
 

+from api.piston_patch import HttpResponseAppRedirect

 from paypal.forms import PaypalSubscriptionForm
 from paypal.rest import execute_agreement, check_agreement, agreement_approval_url, PaypalError
 from paypal.models import BillingAgreement as BillingAgreementModel, BillingPlan
 
 
 from paypal.forms import PaypalSubscriptionForm
 from paypal.rest import execute_agreement, check_agreement, agreement_approval_url, PaypalError
 from paypal.models import BillingAgreement as BillingAgreementModel, BillingPlan
 
 

-@login_required
-def paypal_form(request):
+def paypal_form(request, app=False):

     if request.POST:
     if request.POST:

+        if not request.user.is_authenticated():
+            return HttpResponseForbidden()

         form = PaypalSubscriptionForm(data=request.POST)
         if form.is_valid():
             amount = form.cleaned_data['amount']
             try:
         form = PaypalSubscriptionForm(data=request.POST)
         if form.is_valid():
             amount = form.cleaned_data['amount']
             try:

-                approval_url = agreement_approval_url(amount)
+                approval_url = agreement_approval_url(amount, app=app)

             except PaypalError as e:
                 return render(request, 'paypal/error_page.html', {'error': e.message})
             return HttpResponseRedirect(approval_url)
             except PaypalError as e:
                 return render(request, 'paypal/error_page.html', {'error': e.message})
             return HttpResponseRedirect(approval_url)


@@ -31,7 +33,7 @@ def paypal_form(request):
 
 
 @login_required
 
 
 @login_required

-def paypal_return(request):
+def paypal_return(request, app=False):

     token = request.GET.get('token')
     if not token:
         raise Http404
     token = request.GET.get('token')
     if not token:
         raise Http404


@@ -43,7 +45,15 @@ def paypal_return(request):
             active = check_agreement(resource.id)
             BillingAgreementModel.objects.create(
                 agreement_id=resource.id, user=request.user, plan=plan, active=active, token=token)
             active = check_agreement(resource.id)
             BillingAgreementModel.objects.create(
                 agreement_id=resource.id, user=request.user, plan=plan, active=active, token=token)

-    return render(request, 'paypal/return.html', {'resource': resource})
+    else:
+        resource = None
+    if app:
+        if getattr(resource, 'error'):
+            return HttpResponseAppRedirect('wolnelekturyapp://paypal_error')
+        else:
+            return HttpResponseAppRedirect('wolnelekturyapp://paypal_return')
+    else:
+        return render(request, 'paypal/return.html', {'resource': resource})

 
 
 def paypal_cancel(request):
 
 
 def paypal_cancel(request):