Wire charge-negative-balance to payment methods

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))
-        payments = sum_amounts(Payment.objects.filter(owner=user))
-        return payments - bills
+    bills = sum_amounts(Bill.objects.filter(owner=user))
+    payments = sum_amounts(Payment.objects.filter(owner=user))
+    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