Use ViewSets

Author: WasinTh

account/views.py

@@ -2,6 +2,7 @@
 from django.http import HttpResponse
 from rest_framework.response import Response
 from rest_framework.decorators import api_view
+from rest_framework import viewsets
 from rest_framework import generics
 from account.models import Transaction, Customer
 from account.serializers import TransactionSerializer, CustomerSerializer
@@ -32,3 +33,13 @@ class TransactionView(generics.ListCreateAPIView):
 class CustomerView(generics.ListCreateAPIView):
     queryset = Customer.objects.all()
     serializer_class = CustomerSerializer
+
+
+class TransactionViewSet(viewsets.ModelViewSet):
+    queryset = Transaction.objects.all()
+    serializer_class = TransactionSerializer
+
+
+class CustomerViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = Customer.objects.all()
+    serializer_class = CustomerSerializer

personal_finance/urls.py

@@ -1,8 +1,16 @@
 from django.contrib import admin
 from django.urls import path
-from account.views import current_balance_view, transaction_list_view, TransactionView, CustomerView
+from django.urls import include
+from rest_framework.routers import DefaultRouter
+from account.views import current_balance_view, transaction_list_view, TransactionView, CustomerView, \
+    TransactionViewSet, CustomerViewSet
+
+router = DefaultRouter()
+router.register('account/transaction-viewsets', TransactionViewSet)
+router.register('account/customer-viewsets', CustomerViewSet)
 
 urlpatterns = [
+    path('', include(router.urls)),
     path('admin/', admin.site.urls),
     path('account/current-balance/', current_balance_view),
     path('account/transaction-list/', transaction_list_view),