Django 2.2

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

diff --git a/src/club/payment_methods.py b/src/club/payment_methods.py
index 29221ee..99694ef 100644 (file)
--- a/src/club/payment_methods.py
+++ b/src/club/payment_methods.py
@@ -1,8 +1,12 @@
+# 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
 
 
 class PaymentMethod(object):
+    is_onetime = False
     is_recurring = False
 
     def initiate(self, request, schedule):
@@ -10,6 +14,7 @@ class PaymentMethod(object):
 
 
 class PayU(PaymentMethod):
+    is_onetime = True
     slug = 'payu'
     name = 'PayU'
     template_name = 'club/payment/payu.html'
@@ -30,7 +35,7 @@ class PayU(PaymentMethod):
 
 class PayURe(PaymentMethod):
     slug='payu-re'
-    name = 'PayU Recurring'
+    name = 'PayU (płatność odnawialna)'
     template_name = 'club/payment/payu-re.html'
     is_recurring = True
 
@@ -43,19 +48,24 @@ class PayURe(PaymentMethod):
     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,
-            customer_ip=request.META['REMOTE_ADDR'],
+            customer_ip=ip,
             schedule=schedule,
         )
         return order.put()
         
 
-class PayPalRe(PaymentMethod):
-    slug='paypal-re'
-    name = 'PayPal Recurring'
-    template_name = 'club/payment/paypal-re.html'
+class PayPal(PaymentMethod):
+    slug='paypal'
+    name = 'PayPal (płatność odnawialna)'
+    template_name = 'club/payment/paypal.html'
     is_recurring = True
+    is_onetime = True
 
     def initiate(self, request, schedule):
         return reverse('club_dummy_payment', args=[schedule.key])
@@ -63,13 +73,6 @@ class PayPalRe(PaymentMethod):
 
 methods = []
 
-pos = getattr(settings, 'CLUB_PAYU_POS', None)
-if pos:
-    payu_method = PayU(pos)
-    methods.append(payu_method)
-else:
-    payu_method = None
-
 pos= getattr(settings, 'CLUB_PAYU_RECURRING_POS', None)
 if pos:
     payure_method = PayURe(pos)
@@ -77,8 +80,15 @@ if pos:
 else:
     payure_method = None
 
+pos = getattr(settings, 'CLUB_PAYU_POS', None)
+if pos:
+    payu_method = PayU(pos)
+    methods.append(payu_method)
+else:
+    payu_method = None
+
 
-methods.append(PayPalRe())
+#methods.append(PayPal())
 
 
 method_by_slug = {