Este post é continuação do post Django Rest Framework Requests & Responses.
Finalmente chegamos as views baseadas em classes. A grande vantagem é que com poucas linhas de código já temos nossa API pronta.
Veja como fica a views.py:
from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework import status
from core.models import Person
from core.serializers import PersonSerializerclass PersonList(APIView):
“””
List all persons, or create a new person.
“””
def get(self, request, format=None):
persons = Person.objects.all()
serializer = PersonSerializer(persons, many=True)
return Response(serializer.data)
def post(self, request, format=None):
serializer = PersonSerializer(data=request.data)
if serializer.is_valid():
serializer.save()
return Response(serializer.data, status=status.HTTP_201_CREATED)
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
class PersonDetail(APIView):
“””
Retrieve, update or delete a person instance.
“””
def get_object(self, pk):
try:
return Person.objects.get(pk=pk)
except Person.DoesNotExist:
raise Http404
def get(self, request, pk, format=None):
person = self.get_object(pk)
serializer = PersonSerializer(person)
return Response(serializer.data)
def put(self, request, pk, format=None):
person = self.get_object(pk)
serializer = PersonSerializer(person, data=request.data)
if serializer.is_valid():
serializer.save()
return Response(serializer.data)
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def delete(self, request, pk, format=None):
person = self.get_object(pk)
person.delete()
return Response(status=status.HTTP_204_NO_CONTENT)
E urls.py:
path(‘persons/’, views.PersonList.as_view()),
path(‘persons/<int:pk>/’, views.PersonDetail.as_view()),
]
Usando Mixins
Repare que no exemplo anterior tivemos que definir os métodos get(), post(), put() e delete(). Podemos reduzir ainda mais esse código com o uso de mixins.
from rest_framework import generics
from core.models import Person
from core.serializers import PersonSerializerclass PersonList(mixins.ListModelMixin,
mixins.CreateModelMixin,
generics.GenericAPIView):
queryset = Person.objects.all()
serializer_class = PersonSerializer
def get(self, request, *args, **kwargs):
return self.list(request, *args, **kwargs)
def post(self, request, *args, **kwargs):
return self.create(request, *args, **kwargs)
class PersonDetail(mixins.RetrieveModelMixin,
mixins.UpdateModelMixin,
mixins.DestroyModelMixin,
generics.GenericAPIView):
queryset = Person.objects.all()
serializer_class = PersonSerializer
def get(self, request, *args, **kwargs):
return self.retrieve(request, *args, **kwargs)
def put(self, request, *args, **kwargs):
return self.update(request, *args, **kwargs)
def delete(self, request, *args, **kwargs):
return self.destroy(request, *args, **kwargs)
Usando generic class-based views
E para finalizar usamos ListCreateAPIView e RetrieveUpdateDestroyAPIView que já tem todos os métodos embutidos.
from core.models import Person
from core.serializers import PersonSerializerclass PersonList(generics.ListCreateAPIView):
queryset = Person.objects.all()
serializer_class = PersonSerializer
class PersonDetail(generics.RetrieveUpdateDestroyAPIView):
queryset = Person.objects.all()
serializer_class = PersonSerializer
Versão final de views.py.
Abraços.
Este artigo foi importado automaticamente por fazer parte do Planetário Dev. Quer fazer parte deste HUB de conteúdos? Faça parte do Planetário e veja as vantagens.
Não tem site ou blog? Seja um autor do site e ainda pode ser remunerado.
Leia também: