uncloud/uncloud_pay/selectors.py

32 lines
807 B
Python
Raw Normal View History

2021-01-01 11:41:54 +00:00
from django.utils import timezone
from django.db import transaction
from .models import *
def get_payments_for_user(user):
payments = [ payment.amount for payment in Payment.objects.filter(owner=user) ]
return sum(payments)
def get_spendings_for_user(user):
bills = Bill.objects.filter(owner=user)
2021-01-01 11:41:54 +00:00
amount = 0
for bill in bills:
amount += bill.sum
2021-01-01 11:41:54 +00:00
return amount
@transaction.atomic
def get_balance_for_user(user):
return get_payments_for_user(user) - get_spendings_for_user(user)
2021-01-17 14:53:30 +00:00
@transaction.atomic
def has_enough_balance(user, due_amount):
balance = get_balance_for_user(user)
if balance >= due_amount:
return True
return False
2021-06-20 09:51:27 +00:00
def get_billing_address_for_user(user):
return BillingAddress.objects.filter(owner=user, active=True).first()