Masih melanjutkan dari modul sebelumnya, kali ini kita akan membuat API menggunakan class yang disediakan oleh REST framework.
Masih menggunakan file backend/views.py, berikut code API menggunakan class:
from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework import status
from django.http import Http404
from .models import Article
from .serializers import ArticleSerializer
class ArticleList(APIView):
def get(self, request):
articles = Article.objects.all()
serializer = ArticleSerializer(articles, many=True)
return Response(serializer.data)
def post(self, request):
serializer = ArticleSerializer(data=request.data)
if serializer.is_valid():
serializer.save()
return Response(serializer.data, status=status.HTTP_201_CREATED)
return Response(serializer.error, status=status.HTTP_400_BAD_REQUEST)
class ArticleDetails(APIView):
def get_object(self, slug):
try:
return Article.objects.get(slug=slug)
except Article.DoesNotExist:
raise Http404
def get(self, request, slug):
article = self.get_object(slug)
serializer = ArticleSerializer(article)
return Response(serializer.data)
def put(self, request, slug):
article = self.get_object(slug)
serializer = ArticleSerializer(article, 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, slug):
article = self.get_object(slug)
article.delete()
return Response(status=status.HTTP_204_NO_CONTENT)
Dapat Anda lihat, perbedaanya dengan decorator, API class based mengubah block if menjadi fungsi. Code lebih jelas dan mudah dibaca.
Berikutnya kita perlu menggubah file backend/urls.py:
from django.urls import path
from .views import ArticleList, ArticleDetails
urlpatterns = [
path('articles/', ArticleList.as_view()),
path('articles/<slug:slug>/', ArticleDetails.as_view())
]
Jika dijalankan, sesuai ekspektasi, API tetap berjalan dengan baik.
