Add ADMIN endpoints for bills and payments

uncloud/uncloud_pay/views.py

@@ -6,8 +6,9 @@ from .models import Bill, Payment
 from .serializers import BillSerializer, PaymentSerializer
 from datetime import datetime
 
+###
+# Standard user views:
 
-# to be implemented
 class BalanceViewSet(viewsets.ViewSet):
     # here we return a number
     # number = sum(payments) - sum(bills)
@@ -20,10 +21,9 @@ class BalanceViewSet(viewsets.ViewSet):
     pass
 
 
-class BillViewSet(viewsets.ModelViewSet):
+class BillViewSet(viewsets.ReadOnlyModelViewSet):
     serializer_class = BillSerializer
     permission_classes = [permissions.IsAuthenticated]
-    http_method_names = ['get']
 
     def get_queryset(self):
         return Bill.objects.filter(owner=self.request.user)
@@ -31,10 +31,19 @@ class BillViewSet(viewsets.ModelViewSet):
     def unpaid(self, request):
         return Bill.objects.filter(owner=self.request.user, paid=False)
 
-class PaymentViewSet(viewsets.ModelViewSet):
+class PaymentViewSet(viewsets.ReadOnlyModelViewSet):
+    serializer_class = PaymentSerializer
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get_queryset(self):
+        return Payment.objects.filter(owner=self.request.user)
+
+###
+# Admin views.
+
+class AdminPaymentViewSet(viewsets.ModelViewSet):
     serializer_class = PaymentSerializer
     permission_classes = [permissions.IsAuthenticated]
-    http_method_names = ['get', 'post']
 
     def get_queryset(self):
         return Payment.objects.filter(owner=self.request.user)
@@ -42,7 +51,25 @@ class PaymentViewSet(viewsets.ModelViewSet):
     def create(self, request):
         serializer = self.get_serializer(data=request.data)
         serializer.is_valid(raise_exception=True)
-        serializer.save(owner=request.user,timestamp=datetime.now())
+        serializer.save(timestamp=datetime.now())
+
+        headers = self.get_success_headers(serializer.data)
+        return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
+
+class AdminBillViewSet(viewsets.ModelViewSet):
+    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)
+
+    def create(self, request):
+        serializer = self.get_serializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        serializer.save(created_at=datetime.now())
 
         headers = self.get_success_headers(serializer.data)
         return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)