from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework.authentication import TokenAuthentication, SessionAuthentication
from rest_framework.permissions import IsAuthenticated
from .models import Item, Invoice
from .serializers import ItemSerializer, InvoiceSerializer


class ItemList(APIView):
    authentication_classes = [TokenAuthentication, SessionAuthentication]
    permission_classes = [IsAuthenticated]

    def get(self, request):
        items = Item.objects.filter(pk=Item.invoice)
        serializer = ItemSerializer(items, many=True)
        return Response(serializer.data)


class InvoiceList(APIView):
    authentication_classes = [TokenAuthentication, SessionAuthentication]
    permission_classes = [IsAuthenticated]

    def get(self, request):
        items = Invoice.objects.all()
        serializer = InvoiceSerializer(items, many=True)
        return Response(serializer.data)