您的位置:首页 > 编程语言 > Go语言

DRF Django REST framework 之 视图组件(四)

2019-12-12 17:47 1446 查看

引言

在我们有几十上百的视图类,都有get,post等方法,在功能类似时,会导致大量的重复代码出现,显然还有很多可以优化的地方。这也就有了视图组件,它的功能非常强大,能很好的优化接口逻辑。

视图组件

使用视图组件的mixin优化接口逻辑

  1. 导入 mixins 
  2. 定义序列化类
  3. 定义视图类
# 1.导入mixins
from rest_framework.mixins import (
ListModelMixin,
CreateModelMixin,
DestroyModelMixin,
UpdateModelMixin,
RetrieveModelMixin
)
from rest_framework.generics import GenericAPIView

from DrfOne import models
# 2.定义序列化类
from DrfOne.drf_serializers import BookSerializer

# 3.定义视图类
class BookView(ListModelMixin, CreateModelMixin, GenericAPIView):
# 获取数据源, 固定写法
queryset = models.Book.objects.all()
# 序列化类, 固定写法
serializer_class = BookSerializer
def get(self, request, *args, **kwargs):
return self.list(request, *args, **kwargs)

def post(self, request, *args, **kwargs):
return self.create(request, *args, **kwargs)

class BookFilterView(RetrieveModelMixin, UpdateModelMixin, DestroyModelMixin, GenericAPIView):
# 获取数据源, 固定写法
queryset = models.Book.objects.all()
# 序列化类, 固定写法
serializer_class = BookSerializer

def get(self, request, *args, **kwargs):
return self.retrieve(request, *args, **kwargs)

def put(self, request, *args, **kwargs):
return self.update(request, *args, **kwargs)

def delete(self, request, *args, **kwargs):
return self.destroy(request, *args, **kwargs)

定义序列化类

from rest_framework import serializers

from DrfOne import models

class BookSerializer(serializers.ModelSerializer):
class Meta:
model = models.Book
fields = "__all__"
extra_kwargs = {
# 仅写
"publish": {'write_only': True},
"authors": {'write_only': True},
}

publish_name = serializers.CharField(max_length=32, read_only=True, source="publish.name")
publish_address = serializers.CharField(max_length=32, read_only=True, source="publish.address")
author_name = serializers.SerializerMethodField()

def get_author_name(self, book_obj):
author_list = list()
for author in book_obj.authors.all():
# 注意列表添加字段,author.name而不是author
author_list.append(author.name)
return author_list

注意:操作单条数据的 url 

from django.urls import path, re_path

from DrfOne import views

urlpatterns = [
path('books/', views.BookView.as_view()),
# 需要命名为pk
re_path("books/(?P<pk>\d+)/", views.BookFilterView.as_view()),
]

通过上面代码发现 get , post 等方法内容相似,可以进行再次封装。

使用视图组件的view优化接口逻辑

  1. 导入 generics 
  2. 导入序列化类
  3. 定义视图类

对 mixins 再次优化其余内容不变

# 1.导入generics
from rest_framework import generics

from DrfOne import models
# 2.导入序列化类
from DrfOne.drf_serializers import BookSerializer

# 3.定义视图类
class BookView(generics.ListCreateAPIView):
# 获取数据源, 固定写法
queryset = models.Book.objects.all()
# 序列化类, 固定写法
serializer_class = BookSerializer

class BookFilterView(generics.RetrieveUpdateDestroyAPIView):
queryset = models.Book.objects.all()
serializer_class = BookSerializer

发现还是有重复代码,再次优化,也就是 viewset 。

使用视图组件的viewset优化接口逻辑

看似已经优化的非常完美了,但是在一个对性能要求极高的项目里面,我们的程序还可以继续优化,不断的优化程序是每个程序员必备的技能。

  1. 定义 url 
  2. 导入 viewset 
  3. 导入序列化类
  4. 定义视图类

注意urls.py的变化

from django.urls import path, re_path

from DrfOne import views

urlpatterns = [
# path('books/', views.BookView.as_view()),
# re_path("books/(?P<pk>\d+)/", views.BookFilterView.as_view()),

path("books/", views.BookView.as_view({
"get": "list",
"post": "create",
})),
re_path('books/(?P<pk>\d+)/', views.BookView.as_view({
'get': 'retrieve',
'put': 'update',
'delete': 'destroy'
})),
]

views.py

# 2.导入viewset模块里的ModelViewSet类
from rest_framework.viewsets import ModelViewSet

# 导入应用里的models
from DrfOne import models
# 3.导入序列化类
from DrfOne.drf_serializers import BookSerializer

# 4.定义视图类
class BookView(ModelViewSet):
# 获取数据源, 固定写法
queryset = models.Book.objects.all()
# 序列化类, 固定写法
serializer_class = BookSerializer

~>.<~

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: