Django-RESTful Api之ViewSets和Routers

目的

• **目的：**介绍另一种基于类的视图的写法，它的抽象程度更高，代码量更少

使用ViewSets重构视图

from myApp.models import Student, Grade
from django.contrib.auth.models import User
from myApp.serializers import StudentSerializer, UserSerializer
from rest_framework import generics
from myApp.permissions import IsOwnerOrReadOnly
from rest_framework import permissions
from rest_framework import viewsets

'''
get    /students
post   /students
get    /students/pk
put    /students/pk
delete / students/pk
'''


class StudentViewSet(viewsets.ModelViewSet):
    queryset = Student.objects.all()
    serializer_class = StudentSerializer
    permission_classes = (permissions.IsAuthenticatedOrReadOnly, IsOwnerOrReadOnly)

    def perform_create(self, serializer):
        serializer.save(owner=self.request.user)

class UserViewSet(viewsets.ReadOnlyModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer

重构路由

from django.conf.urls import url, include
from myApp.views import StudentViewSet, UserViewSet
from rest_framework.urlpatterns import format_suffix_patterns

student_list = StudentViewSet.as_view({
    'get': 'list',
    'post': 'create',
})
student_detail = StudentViewSet.as_view({
    'get': 'retrieve',
    'put': 'update',
    'patch': 'partial_update',
    'delete': 'destroy',
})

user_list = UserViewSet.as_view({
    'get': 'list',
})
user_detail = UserViewSet.as_view({
    'get': 'retrieve',
})

urlpatterns = format_suffix_patterns([
    url(r'^students/$', student_list, name='studentList'),
    url(r'^students/(?P<pk>\d+)', student_detail, name='studentDetail'),
    url(r'^users/$', user_list, name='userList'),
    url(r'^users/(?P<pk>\d+)', user_detail, name='userDetail'),
])

使用Routers

from django.conf.urls import url, include
from myApp import views
from rest_framework.routers import DefaultRouter

router = DefaultRouter()
router.register(r'students', views.StudentViewSet)
router.register(r'users', views.UserViewSet)

urlpatterns = [
    url(r'^', include(router.urls))
]