Django-RESTful Api 之 ViewSets 和 Routers

发表于 更新于 分类于 > > 本文字数： 2.2k 阅读时长 ≈ 4 分钟

目的

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

使用 ViewSets 重构视图

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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)
1
2
3
class UserViewSet(viewsets.ReadOnlyModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer

重构路由

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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

1
2
3
4
5
6
7
8
9
10
11
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))
]