from rest_framework.viewsets import GenericViewSet
from rest_framework.response import Response
from rest_framework import status


from .models import Team, SupportInfo
from .serializers import TeamSerializer, SupportInfoSerializer

        
class TeamViewSet(GenericViewSet):
    queryset = Team
    serializer_class = TeamSerializer

    def list(self, request):
        queryset = self.get_queryset().objects.all()
        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data, status=status.HTTP_200_OK)
    
class SupportInfoViewSet(GenericViewSet):
    queryset = SupportInfo
    serializer_class = SupportInfoSerializer

    def list(self, request):
        queryset = self.get_queryset().objects.all()
        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data, status=status.HTTP_200_OK)