Wire charge-negative-balance to payment methods

uncloud/uncloud/urls.py

@@ -30,6 +30,7 @@ router.register(r'vm/vm', vmviews.VMProductViewSet, basename='vmproduct')
 
 # Pay
 router.register(r'user', payviews.UserViewSet, basename='user')
+router.register(r'payment-method', payviews.PaymentMethodViewSet, basename='payment-method')
 router.register(r'bill', payviews.BillViewSet, basename='bill')
 router.register(r'order', payviews.OrderViewSet, basename='order')
 router.register(r'payment', payviews.PaymentViewSet, basename='payment')

uncloud/uncloud_pay/helpers.py

@@ -1,12 +1,20 @@
 from functools import reduce
-from .models import Bill, Payment
+from .models import Bill, Payment, PaymentMethod
 
 def sum_amounts(entries):
-        return reduce(lambda acc, entry: acc + entry.amount, entries, 0)
+    return reduce(lambda acc, entry: acc + entry.amount, entries, 0)
 
 
 def get_balance_for(user):
-        bills = sum_amounts(Bill.objects.filter(owner=user))
+    bills = sum_amounts(Bill.objects.filter(owner=user))
-        payments = sum_amounts(Payment.objects.filter(owner=user))
+    payments = sum_amounts(Payment.objects.filter(owner=user))
-        return payments - bills
+    return payments - bills
 
+def get_payment_method_for(user):
+    methods = PaymentMethod.objects.filter(owner=user)
+    for method in methods:
+        # Do we want to do something with non-primary method?
+        if method.primary:
+            return method
+
+    return None

uncloud/uncloud_pay/management/commands/charge-negative-balance.py

@@ -1,7 +1,7 @@
 from django.core.management.base import BaseCommand
 from uncloud_auth.models import User
 from uncloud_pay.models import Order, Bill
-from uncloud_pay.helpers import get_balance_for
+from uncloud_pay.helpers import get_balance_for, get_payment_method_for
 
 from datetime import timedelta
 from django.utils import timezone
@@ -19,5 +19,14 @@ class Command(BaseCommand):
             balance = get_balance_for(user)
             if balance < 0:
                 print("User {} has negative balance ({}), charging.".format(user.username, balance))
-                # TODO: charge
+                payment_method = get_payment_method_for(user)
+                if payment_method != None:
+                    amount_to_be_charged = abs(balance)
+                    charge_ok = payment_method.charge(amount_to_be_charged)
+                    if not charge_ok:
+                        print("ERR: charging {} with method {} failed"
+                                .format(user.username, payment_method.uuid)
+                                )
+                else:
+                    print("ERR: no payment method registered for {}".format(user.username))
         print("=> Done.")