美文网首页django-rest-framework开发
django-rest-framework(概念篇)——Mixi

django-rest-framework(概念篇)——Mixi

作者: Ccccolin_aha | 来源:发表于2018-04-09 10:54 被阅读24次

    from rest_framework import viewsets这个viewset中,只要有5类Minxin,他们与http方法对应如下:

    mixins
      1. CreateModelMixin
    # 源码
    class CreateModelMixin(object):
        """
        Create a model instance ==>创建一个实例
        """
        def create(self, request, *args, **kwargs):
        
        # 获取相关serializer
            serializer = self.get_serializer(data=request.data)
            
            # 进行serializer的验证
            # raise_exception=True,一旦验证不通过,不再往下执行,直接引发异常
            serializer.is_valid(raise_exception=True)
            
            # 调用perform_create()方法,保存实例
            self.perform_create(serializer)
            
            headers = self.get_success_headers(serializer.data)
            return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
    
        def perform_create(self, serializer):
        # 保存实例
            serializer.save()
    
        def get_success_headers(self, data):
            try:
                return {'Location': str(data[api_settings.URL_FIELD_NAME])}
            except (TypeError, KeyError):
                return {}
    
    CreateModelMixin

    由上图可以看出这个类的一个逻辑,其中,perform_create( )对serializer直接进行save保存,当在一些情境下,我们需要对perform_create( )进行重写。
    那么我们现在可能有一个下面的需要:
      假设现在有一个course课程model,里面维持了一个数,记录课程收藏数,还存在一个用户收藏userfav的model(应当有一个外键指向course),当一个用户对课程进行收藏,理论上现在post进来的应该是userfav的instance,显然,我们还需要对相应course的收藏数进行+1。
      这个时候,我们就需要重写perform_create( )方法!

    def perform_create(self, serializer):
    # 重写save的逻辑
        instance = serializer.save()
        course = instance.course
        course.fav_num += 1
        course.save()
    
      1. ListModelMixin
    # 源码
    class RetrieveModelMixin(object):
        """
        Retrieve a model instance.==> 获取某一个对象的具体信息
        """
        def retrieve(self, request, *args, **kwargs):
            # 一般访问的url都为/obj/id/这种新式
            # get_object()可以获取到这个id的对象
            # 注意在viewset中设置lookup_field获取重写get_object()方法可以指定id具体对象是什么~!
            instance = self.get_object()
            serializer = self.get_serializer(instance)
            return Response(serializer.data)
    

    对retrieve这个方法的重写几率比较高,例如我们在增加点击数的时候,经常要对其进行一个重写。

      1. RetrieveModelMixin
    # 源码
    class UpdateModelMixin(object):
        """
        Update a model instance.==> 更新某个具体对象的内容
        """
        def update(self, request, *args, **kwargs):
            partial = kwargs.pop('partial', False)
            instance = self.get_object()
            serializer = self.get_serializer(instance, data=request.data, partial=partial)
            serializer.is_valid(raise_exception=True)
            self.perform_update(serializer)
    
            if getattr(instance, '_prefetched_objects_cache', None):
                # If 'prefetch_related' has been applied to a queryset, we need to
                # forcibly invalidate the prefetch cache on the instance.
                instance._prefetched_objects_cache = {}
    
            return Response(serializer.data)
    
        def perform_update(self, serializer):
            serializer.save()
    
        def partial_update(self, request, *args, **kwargs):
            kwargs['partial'] = True
            return self.update(request, *args, **kwargs)
    

    RetrieveModelMixin的实现逻辑基本整合了Create以及Retrieve,先得到具体的实例,再对其进行验证以及保存,如果需要对更新这个逻辑进行自定义,那么需要重写perform_update( )方法,而尽量少去重写update( )

      1. DestroyModelMixin
    # 源码
    class DestroyModelMixin(object):
        """
        Destroy a model instance.
        """
        def destroy(self, request, *args, **kwargs):
            instance = self.get_object()
            self.perform_destroy(instance)
            return Response(status=status.HTTP_204_NO_CONTENT)
    
        def perform_destroy(self, instance):
            instance.delete()
    

    DestroyModelMixin的逻辑也相对比较简单,我们取CreateModelMixin下面的例子,当我们取消收藏,那么我们的DestroyModelMixin就发挥作用了。同理:

    def perform_create(self, serializer):
        instance = serializer.save()
        course = instance.course
        if course.fav_num > 0:
            course.fav_num -= 1
        else:
            course.fav_num = 0
        course.save()
    

    一般情况下,当我们在操作某一个model的时候,涉及到另外一个model中数据的修改,那么就需要对这个mixins下执行save的逻辑的方法进行重写。

    相关文章

      网友评论

        本文标题:django-rest-framework(概念篇)——Mixi

        本文链接:https://www.haomeiwen.com/subject/hbgphftx.html