fix

[wolnelektury.git] / src / club / payment_methods.py

diff --git a/src/club/payment_methods.py b/src/club/payment_methods.py
index b215b1d..0c187a8 100644 (file)
--- a/src/club/payment_methods.py
+++ b/src/club/payment_methods.py
@@ -1,10 +1,17 @@
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#

 from django.conf import settings
 from django.urls import reverse
 from django.conf import settings
 from django.urls import reverse

+from paypal.rest import agreement_approval_url

 
 
 class PaymentMethod(object):
     is_onetime = False
     is_recurring = False
 
 
 class PaymentMethod(object):
     is_onetime = False
     is_recurring = False

+    expiration_reliable = False
+    cancellable = False
+    updateable = False

 
     def initiate(self, request, schedule):
         return reverse('club_dummy_payment', args=[schedule.key])
 
     def initiate(self, request, schedule):
         return reverse('club_dummy_payment', args=[schedule.key])


@@ -12,6 +19,7 @@ class PaymentMethod(object):
 
 class PayU(PaymentMethod):
     is_onetime = True
 
 class PayU(PaymentMethod):
     is_onetime = True

+    expiration_reliable = True

     slug = 'payu'
     name = 'PayU'
     template_name = 'club/payment/payu.html'
     slug = 'payu'
     name = 'PayU'
     template_name = 'club/payment/payu.html'


@@ -31,10 +39,13 @@ class PayU(PaymentMethod):
 
 
 class PayURe(PaymentMethod):
 
 
 class PayURe(PaymentMethod):

-    slug='payu-re'
-    name = 'PayU Recurring'
+    slug = 'payu-re'
+    name = 'PayU recurring'

     template_name = 'club/payment/payu-re.html'
     is_recurring = True
     template_name = 'club/payment/payu-re.html'
     is_recurring = True

+    expiration_reliable = True
+    cancellable = True
+    updateable = True

 
     def __init__(self, pos_id):
         self.pos_id = pos_id
 
     def __init__(self, pos_id):
         self.pos_id = pos_id


@@ -45,46 +56,63 @@ class PayURe(PaymentMethod):
     def pay(self, request, schedule):
         # Create order, put it and see what happens next.
         from .models import PayUOrder
     def pay(self, request, schedule):
         # Create order, put it and see what happens next.
         from .models import PayUOrder

+        if request is not None:
+            ip = request.META['REMOTE_ADDR']
+        else:
+            ip = '127.0.0.1'

         order = PayUOrder.objects.create(
             pos_id=self.pos_id,
         order = PayUOrder.objects.create(
             pos_id=self.pos_id,

-            customer_ip=request.META['REMOTE_ADDR'],
+            customer_ip=ip,

             schedule=schedule,
         )
         return order.put()
         
 
 class PayPal(PaymentMethod):
             schedule=schedule,
         )
         return order.put()
         
 
 class PayPal(PaymentMethod):

-    slug='paypal'
+    slug = 'paypal'

     name = 'PayPal'
     template_name = 'club/payment/paypal.html'
     is_recurring = True
     name = 'PayPal'
     template_name = 'club/payment/paypal.html'
     is_recurring = True

-    is_onetime = True
+    is_onetime = False

 
     def initiate(self, request, schedule):
 
     def initiate(self, request, schedule):

-        return reverse('club_dummy_payment', args=[schedule.key])
+        app = request.GET.get('app')
+        return agreement_approval_url(schedule.amount, schedule.key, app=app)
+
+    def pay(self, request, schedule):
+        from datetime import date, timedelta, datetime
+        from pytz import utc
+        tomorrow = datetime(*(date.today() + timedelta(2)).timetuple()[:3], tzinfo=utc)
+        any_active = False
+        for ba in schedule.billingagreement_set.all():
+            active = ba.check_agreement()
+            ba.active = active
+            ba.save()
+            if active:
+                any_active = True
+        if any_active:
+            schedule.expires_at = tomorrow
+            schedule.save()

 
 
 methods = []
 
 
 
 methods = []
 

-pos = getattr(settings, 'CLUB_PAYU_POS', None)
+pos = getattr(settings, 'CLUB_PAYU_RECURRING_POS', None)

 if pos:
 if pos:

-    payu_method = PayU(pos)
-    methods.append(payu_method)
+    recurring_payment_method = PayURe(pos)
+    methods.append(recurring_payment_method)

 else:
 else:

-    payu_method = None
+    recurring_payment_method = None

 
 

-pos= getattr(settings, 'CLUB_PAYU_RECURRING_POS', None)
+pos = getattr(settings, 'CLUB_PAYU_POS', None)

 if pos:
 if pos:

-    payure_method = PayURe(pos)
-    methods.append(payure_method)
+    single_payment_method = PayU(pos)
+    methods.append(single_payment_method)

 else:
 else:

-    payure_method = None
-
+    single_payment_method = None

 
 

-methods.append(PayPal())

 
 
 
 

-method_by_slug = {
-    m.slug: m
-    for m in methods
-}
+methods.append(
+    PayPal()
+)