15 Django REST Framework 給api添加自定義搜索條件

1、ListModelMixin源碼

# 源碼

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

2、解決方法：

01-增長篩選條件 - 重寫ListModelMixin的list方法

注：不會改變原有的篩選條件的方式 ：queryset = self.filter_queryset(self.queryset)；

　　若是放棄原有的篩選條件的方式： queryset = self.queryset  都從新作篩選。

def list(self, request, *args, **kwargs):
    queryset = self.filter_queryset(self.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)

示例：

class RoomReservationViewSet(viewsets.ModelViewSet):
    """
        list:
            返回列出全部 會議室預約 數據.
            檢索方式：
              官網地址：https://github.com/miki725/django-url-filter
                  單條件檢索：/api/?work_status=已完成&&work_type=維修
                  單字段多條件檢索： /api/?work_status__in=已完成,新建
                  非字段檢索： /api/?work_status__icontains!=待審批
                  外鍵字段檢索：/api/?approval_team__name=審批組
                  時間檢索: start_time__range
                      /api/office_room/?start_time__range=2019-04-11,2019-04-12
                      /api/office_room/?start_time__year=2019
                      /api/office_room/?start_time__gt=2019-04-11

        create:
            建立一條 會議室預約 數據.

        retrieve:
            返回一個 會議室預約 實例.

        update:
            更新一條 會議室預約 數據.

        partial_update:
            更新 會議室預約 的部分字段.

        delete:
            刪除一條 會議室預約 數據.
    """
    queryset = models.RoomReservation.objects.all().order_by('-riid')
    pagination_class = StandardResultsSetPagination
    serializer_class = RoomReservationSerializer
    # 使用過濾器
    filter_backends = (DjangoFilterBackend, filters.SearchFilter,)

    authentication_classes = ()
    permission_classes = (AllowAny, )
    # 定義須要使用過濾器的字段
    filter_fields = ('riid', 'roomid', 'start_time', 'end_time', 'res_user', 'com_name', 'room_status', 'res_time')

    search_fields = ('riid', 'roomid', 'start_time', 'end_time', 'res_user', 'com_name', 'room_status', 'res_time')

    def list(self, request, *args, **kwargs):
        from rest_framework import mixins
        # queryset = mixins.ListModelMixin.list(request, *args, **kwargs).queryset
        time_now = datetime.datetime.now()
        room_obj = RoomReservation.objects.all()
        # 更新 會議室狀態
        for i in room_obj:
            if i.start_time < time_now < i.end_time:
                i.room_status = '使用中'
                i.save()
            elif time_now > i.end_time:
                i.room_status = '已結束'
                i.save()

        room_num = request.GET.get('room_num')
        start_time = request.GET.get('start_time')
        end_time = request.GET.get('end_time')
        room_id = request.GET.get('room_id')
        room_status = request.GET.get('room_status')
        # queryset = self.queryset
        queryset = self.filter_queryset(self.queryset)
        if room_id:
            queryset = models.RoomReservation.objects.filter(roomid__room_num=room_id).order_by('-riid')
        if room_status:
            queryset = models.RoomReservation.objects.filter(room_status=room_status).order_by('-riid')

        room_num_obj = models.RoomNumber.objects.filter(room_num=room_num).first()
        # lt 小於  gt 大於
        if room_num_obj:
            queryset = models.RoomReservation.objects.filter(roomid=room_num_obj).order_by('-riid')
        if room_num_obj and start_time and end_time:
            queryset = models.RoomReservation.objects.filter(roomid=room_num_obj).filter(start_time__lt=end_time).filter(end_time__gt=start_time).order_by('-riid')

        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)

    def perform_create(self, serializer):
        instance = serializer.save()
        instance.res_time = datetime.datetime.now()
        instance.save()

views.py

 02-重寫filter_queryset方法

注意：此處 worksheet_no 字段 爲 外鍵字段；

def filter_queryset(self, queryset):
    queryset = super(TasksViewSet, self).filter_queryset(self.queryset)
    worksheet_no = self.request.query_params.get('worksheet_no', None)
    print('worksheet_no', worksheet_no)
    if worksheet_no:
        worksheet_no_instance = WorkSheet.objects.filter(work_no=worksheet_no).first()
        print('worksheet_no_instance', worksheet_no_instance)
        queryset = self.queryset.filter(worksheet_no=worksheet_no_instance)

    return queryset