Wire charge-negative-balance to payment methods

This commit is contained in:
fnux 2020-02-28 09:10:36 +01:00
parent 059791e2f2
commit 4bed53c8a8
3 changed files with 25 additions and 7 deletions

View file

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

View file

@ -1,12 +1,20 @@
from functools import reduce from functools import reduce
from .models import Bill, Payment from .models import Bill, Payment, PaymentMethod
def sum_amounts(entries): 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): 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

View file

@ -1,7 +1,7 @@
from django.core.management.base import BaseCommand from django.core.management.base import BaseCommand
from uncloud_auth.models import User from uncloud_auth.models import User
from uncloud_pay.models import Order, Bill 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 datetime import timedelta
from django.utils import timezone from django.utils import timezone
@ -19,5 +19,14 @@ class Command(BaseCommand):
balance = get_balance_for(user) balance = get_balance_for(user)
if balance < 0: if balance < 0:
print("User {} has negative balance ({}), charging.".format(user.username, balance)) 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.") print("=> Done.")