CreateAPIView, ListAPIView, RetrieveAPIView, DestroyAPIView UpdateAPIView重构方法

ListAPIView

'''
重构list方法，

def list(self, request, *args, **kwargs):
    response = super().list(request, *args, **kwargs)
    
response 父类list方法的返回值Response(serializer.data),即序列化后的数据serializer.data

response.data = context 给response添加键值对

return response 再将新的response返回给前端
'''

父类list方法

class ListModelMixin:
    """
    List a queryset.
    """
    def list(self, request, *args, **kwargs):
        queryset = self.filter_queryset(self.get_queryset())

        page = self.paginate_queryset(queryset)
        if page is not None:
            serializer = self.get_serializer(page, many=True)
            return self.get_paginated_response(serializer.data)

        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data)

重构list方法

class BidView(ListAPIView, CreateAPIView):

    authentication_classes = [UserAuthentication, ]
    permission_classes = [BidPermission, ]
    queryset = models.BidRecord.objects.all().order_by('-id')
    serializer_class = BidModelSerializer
    filter_backends = [BidItemFilter, ]

    def list(self, request, *args, **kwargs):
        response = super().list(request, *args, **kwargs)
        if response.status_code != status.HTTP_200_OK:
            return response
        lot_id = request.query_params.get('lot_id')
        lot_object = models.Lot.objects.filter(id=lot_id).first()

        context = {
            'unit': lot_object.unit,
            'price': response.data[0]['price'] if response.data else lot_object.start_price,
            'bid_list': response.data
        }
        response.data = context
        return response

CreateAPIView

'''
重构perform_create方法

前端数据校验成功后，执行perform_create(self, serializer)方法保存数据到数据库

'''

父类list方法

class CreateModelMixin:
    """
    Create a model instance.
    """
    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        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 {}

重构perform_create方法

class BidView(ListAPIView, CreateAPIView):

    def perform_create(self, serializer):
        serializer.save(user_id=self.request.user.id, status=1)

RetrieveAPIView

'''
重构get方法/retrieve方法

def get(self, request, *args, **kwargs):
    # 1. 获取详细信息
    response = self.retrieve(request, *args, **kwargs)

同ListAPIView相似

'''

父类方法

class RetrieveAPIView(mixins.RetrieveModelMixin,
                  GenericAPIView):
    """
    Concrete view for retrieving a model instance.
    """
    def get(self, request, *args, **kwargs):
        return self.retrieve(request, *args, **kwargs)


class RetrieveModelMixin:
    """
    Retrieve a model instance.
    """
    def retrieve(self, request, *args, **kwargs):
        instance = self.get_object()
        serializer = self.get_serializer(instance)
        return Response(serializer.data)

重构get方法/retrieve方法

class NewsDetailView(RetrieveAPIView):
    """
    获取动态详细接口
    """
    queryset = models.News.objects
    serializer_class = RetrieveNewsDetailModelSerializerSerializer

    def get(self, request, *args, **kwargs):
        # 1. 获取详细信息
        response = self.retrieve(request, *args, **kwargs)

        # 2. 处理用户浏览记录，当前用户添加到记录中
        #    2.1 用户未登录，不记录
        #    2.2 用户登录，已记录则不再记录记录，未记录则添加到浏览记录中。
        if not request.user:
            return response
        news_object = self.get_object()
        if not news_object:
            return response

        viewer_queryset = models.ViewerRecord.objects.filter(news=news_object, user_id=request.user.id)
        if not viewer_queryset.exists():
            models.ViewerRecord.objects.create(news=news_object, user_id=request.user.id)
            models.News.objects.filter(id=news_object.id).update(favor_count=F("viewer_count") + 1)

        return response

DestroyAPIView

'''
啦啦啦，没用过。。。，用过补充

'''

父类方法

class DestroyModelMixin:
    """
    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()

UpdateAPIView

'''
啦啦啦，没用过。。。，用过补充

'''

父类方法

class UpdateAPIView(mixins.UpdateModelMixin,
                    GenericAPIView):
    """
    Concrete view for updating a model instance.
    """
    def put(self, request, *args, **kwargs):
        return self.update(request, *args, **kwargs)

    def patch(self, request, *args, **kwargs):
        return self.partial_update(request, *args, **kwargs)


class UpdateModelMixin:
    """
    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)