Masih membahas penggunaan ViewSet, kali ini kita akan membahas model viewset.
Buka file backend/views.py, gunakan kode berikut:
from .models import Article
from .serializers import ArticleSerializer
from rest_framework import viewsets
class ArticleViewSet(viewsets.ModelViewSet):
queryset = Article.objects.all()
serializer_class = ArticleSerializer
lookup_field = 'slug'
Dapat Anda lihat, kode semakin pendek dan clean. Untuk backend/urls.py juga tidak ada perubahan dari kode modul sebelumnya.
from django.urls import path, include
from .views import ArticleViewSet
from rest_framework.routers import DefaultRouter
router = DefaultRouter()
router.register('articles', ArticleViewSet, basename="articles")
urlpatterns = [
path('', include(router.urls))
]
Jika Anda buka melalui browser, sesuai ekspektasi, API berjalan dengan baik.
