implement balance getting

uncloud_pay/views.py

@@ -24,12 +24,15 @@ import logging
 
 from .models import *
 from .serializers import *
+from .selectors import *
+
 from datetime import datetime
 from vat_validator import sanitize_vat
 import uncloud_pay.stripe as uncloud_stripe
 
 logger = logging.getLogger(__name__)
 
+
 ###
 # 2020-12 checked code
 
@@ -75,6 +78,16 @@ class PaymentViewSet(viewsets.ModelViewSet):
     def get_queryset(self):
         return Payment.objects.filter(owner=self.request.user)
 
+class BalanceViewSet(viewsets.ViewSet):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def list(self, request):
+        serializer = BalanceSerializer(data={
+            'balance': get_balance_for_user(self.request.user)
+        })
+        serializer.is_valid()
+        return Response(serializer.data)
+
 
 ###
 # Payments and Payment Methods.