Initial stripe playground

This commit is contained in:
fnux 2020-03-02 22:26:40 +01:00
commit 4ad737ed90
7 changed files with 124 additions and 3 deletions

View file

@ -57,9 +57,15 @@ class UserViewSet(viewsets.ReadOnlyModelViewSet):
return get_user_model().objects.all()
class PaymentMethodViewSet(viewsets.ModelViewSet):
serializer_class = PaymentMethodSerializer
permission_classes = [permissions.IsAuthenticated]
def get_serializer_class(self):
if self.action == 'create':
return CreatePaymentMethodSerializer
else:
return PaymentMethodSerializer
def get_queryset(self):
return PaymentMethod.objects.filter(owner=self.request.user)