Cleanup/reorder uncloud_pay views

uncloud/uncloud_pay/models.py

@@ -171,7 +171,7 @@ class PaymentMethod(models.Model):
         pass
 
 ###
-# Bills & Payments.
+# Bills.
 
 class Bill(models.Model):
     uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)

uncloud/uncloud_pay/views.py

@@ -16,40 +16,7 @@ from datetime import datetime
 import uncloud_pay.stripe as uncloud_stripe
 
 ###
-# Standard user views:
+# Users.
-
-class BalanceViewSet(viewsets.ViewSet):
-    # here we return a number
-    # number = sum(payments) - sum(bills)
-
-    #bills = Bill.objects.filter(owner=self.request.user)
-    #payments = Payment.objects.filter(owner=self.request.user)
-
-    # sum_paid = sum([ amount for amount payments..,.  ]) # you get the picture
-    # sum_to_be_paid = sum([ amount for amount bills..,.  ]) # you get the picture
-    pass
-
-
-class BillViewSet(viewsets.ReadOnlyModelViewSet):
-    serializer_class = BillSerializer
-    permission_classes = [permissions.IsAuthenticated]
-
-    def get_queryset(self):
-        return Bill.objects.filter(owner=self.request.user)
-
-class PaymentViewSet(viewsets.ReadOnlyModelViewSet):
-    serializer_class = PaymentSerializer
-    permission_classes = [permissions.IsAuthenticated]
-
-    def get_queryset(self):
-        return Payment.objects.filter(owner=self.request.user)
-
-class OrderViewSet(viewsets.ReadOnlyModelViewSet):
-    serializer_class = OrderSerializer
-    permission_classes = [permissions.IsAuthenticated]
-
-    def get_queryset(self):
-        return Order.objects.filter(owner=self.request.user)
 
 class UserViewSet(viewsets.ReadOnlyModelViewSet):
     serializer_class = UserSerializer
@@ -58,6 +25,16 @@ class UserViewSet(viewsets.ReadOnlyModelViewSet):
     def get_queryset(self):
         return get_user_model().objects.all()
 
+###
+# Payments and Payment Methods.
+
+class PaymentViewSet(viewsets.ReadOnlyModelViewSet):
+    serializer_class = PaymentSerializer
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get_queryset(self):
+        return Payment.objects.filter(owner=self.request.user)
+
 class PaymentMethodViewSet(viewsets.ModelViewSet):
     permission_classes = [permissions.IsAuthenticated]
 
@@ -149,9 +126,30 @@ class PaymentMethodViewSet(viewsets.ModelViewSet):
             error = 'Could not fetch payment method from stripe. Please try again.'
             return Response({'error': error})
 
+###
+# Bills and Orders.
+
+class BillViewSet(viewsets.ReadOnlyModelViewSet):
+    serializer_class = BillSerializer
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get_queryset(self):
+        return Bill.objects.filter(owner=self.request.user)
+
+    def unpaid(self, request):
+        return Bill.objects.filter(owner=self.request.user, paid=False)
+
+
+class OrderViewSet(viewsets.ReadOnlyModelViewSet):
+    serializer_class = OrderSerializer
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get_queryset(self):
+        return Order.objects.filter(owner=self.request.user)
+
 
 ###
-# Admin views.
+# Old admin stuff.
 
 class AdminPaymentViewSet(viewsets.ModelViewSet):
     serializer_class = PaymentSerializer