Пример #1
0
    def get_filter_class(self, view, queryset=None):
        """
        Return the django-filters `FilterSet` used to filter the queryset.
        """
        filter_class = getattr(view, 'filter_class', None)
        filter_fields = getattr(view, 'filter_fields', None)

        new_filter_fields = []
        if filter_fields:
            for field in filter_fields:
                if settings.REST_FRAMEWORK["INITCAP_FIELDS"]:
                    new_filter_fields.append(unconvert(field))
                else:
                    new_filter_fields.append(field)
        
        if filter_class:
            filter_model = filter_class.Meta.model

            assert issubclass(filter_model, queryset.model), \
                'FilterSet model %s does not match queryset model %s' % \
                (filter_model, queryset.model)

            return filter_class
        
        if filter_fields:
            class AutoFilterSet(self.default_filter_set):
                class Meta:
                    model = queryset.model
                    fields = new_filter_fields
                    order_by = True
            return AutoFilterSet


        return None
Пример #2
0
    def filter_queryset(self, request, queryset, view):
        search_fields = getattr(view, 'search_fields', None)
        
        new_search_fields = []
        if search_fields:
            for field in search_fields:
                if settings.REST_FRAMEWORK["INITCAP_FIELDS"]:
                    new_search_fields.append(unconvert(field))
                else:
                    new_search_fields.append(field)
        
        
        if not new_search_fields:
            return queryset

        orm_lookups = [self.construct_search(str(search_field))
                       for search_field in new_search_fields]

        for search_term in self.get_search_terms(request):
            or_queries = [models.Q(**{orm_lookup: search_term})
                          for orm_lookup in orm_lookups]
            queryset = queryset.filter(reduce(operator.or_, or_queries))

        return queryset