Move bill generation logic to Bill class, initial work for prepaid

uncloud/uncloud_pay/management/commands/handle-overdue-bills.py

@@ -1,12 +1,12 @@
 from django.core.management.base import BaseCommand
 from uncloud_auth.models import User
-from uncloud_pay.models import Order, Bill, get_balance_for
+from uncloud_pay.models import Bill
 
 from datetime import timedelta
 from django.utils import timezone
 
 class Command(BaseCommand):
-    help = 'Generate bills and charge customers if necessary.'
+    help = 'Take action on overdue bills.'
 
     def add_arguments(self, parser):
         pass
@@ -15,28 +15,9 @@ class Command(BaseCommand):
         users = User.objects.all()
         print("Processing {} users.".format(users.count()))
         for user in users:
-            balance = get_balance_for(user)
-            if balance < 0:
-                print("User {} has negative balance ({}), checking for overdue bills."
-                        .format(user.username, balance))
-
-                # Get bills DESCENDING by creation date (= latest at top).
-                bills = Bill.objects.filter(
-                        owner=user,
-                        due_date__lt=timezone.now()
-                        ).order_by('-creation_date')
-                overdue_balance = abs(balance)
-                overdue_bills = []
-                for bill in bills:
-                    if  overdue_balance < 0:
-                        break # XXX: I'm (fnux) not fond of breaks!
-
-                    overdue_balance -= bill.amount
-                    overdue_bills.append(bill)
-
-                for bill in overdue_bills:
-                    print("/!\ Overdue bill for {}, {} with amount {}"
-                            .format(user.username, bill.uuid, bill.amount))
-                    # TODO: take action?
+            for bill in Bill.get_overdue_for(user):
+                print("/!\ Overdue bill for {}, {} with amount {}"
+                        .format(user.username, bill.uuid, bill.amount))
+                # TODO: take action?
 
         print("=> Done.")