from rest_framework import generics, views, status, permissions, parsers
from rest_framework.response import Response

from core.apps.contracts.serializers import contract as contract_serializer
from core.apps.contracts.models.contract import Contract


class ContractCreateApiView(generics.CreateAPIView):
    serializer_class = contract_serializer.ContractCreateSerializer
    queryset = Contract.objects.all()
    permission_classes = [permissions.IsAuthenticated]
    parser_classes = [parsers.MultiPartParser, parsers.FormParser] 
    
    def get_serializer_context(self):
        return {'user': self.request.user}
    

class ContractListApiView(generics.ListAPIView):
    serializer_class = contract_serializer.ContractListSerializer
    queryset = Contract.objects.all()

    def get_queryset(self):
        return super().get_queryset()