Add initial structure for payment methods

uncloud/uncloud_pay/views.py

@@ -4,8 +4,8 @@ from rest_framework import viewsets, permissions, status
 from rest_framework.response import Response
 from rest_framework.decorators import action
 
-from .models import Bill, Payment, Order
-from .serializers import BillSerializer, PaymentSerializer, UserSerializer, OrderSerializer
+from .models import *
+from .serializers import *
 from datetime import datetime
 
 ###
@@ -58,6 +58,21 @@ class UserViewSet(viewsets.ReadOnlyModelViewSet):
     def balance(self, request):
         return Response(status=status.HTTP_204_NO_CONTENT)
 
+class PaymentMethodViewSet(viewsets.ModelViewSet):
+    serializer_class = PaymentMethodSerializer
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get_queryset(self):
+        return PaymentMethod.objects.filter(owner=self.request.user)
+
+    def create(self, request):
+        serializer = self.get_serializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        serializer.save(owner=request.user)
+
+        headers = self.get_success_headers(serializer.data)
+        return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
+
 ###
 # Admin views.