25 lines
879 B
Python
25 lines
879 B
Python
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.objects.all()
|
|
serializer_class = SupportInfoSerializer
|
|
|
|
def list(self, request):
|
|
queryset = self.get_queryset().first()
|
|
serializer = self.get_serializer(queryset)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|