diff --git a/payment/views.py b/payment/views.py index 30709fa..722b008 100644 --- a/payment/views.py +++ b/payment/views.py @@ -1,5 +1,6 @@ from rest_framework.views import APIView from rest_framework.response import Response +from rest_framework.generics import ListAPIView from rest_framework import status import stripe from django.conf import settings @@ -11,10 +12,21 @@ import json from django.http import HttpResponse from company.models import Company +from .serializers import ProductSerializer stripe.api_key = settings.STRIPE_SECRET_KEY +class ProductListView(ListAPIView): + queryset = Product.objects.all() + serializer_class = ProductSerializer + permission_classes = [IsAuthenticated] + + def get_queryset(self): + queryset = super().get_queryset() + return queryset + + class StripeCheckoutSessionView(APIView): permission_classes = [IsAuthenticated]