Python 用DRF进行IDs过滤的一般方法

在使用DRF时，是否有一种通用的方法可以通过ID数组进行过滤？ 例如，如果我想返回具有以下ID的所有图像，我会这样做：

/images/？ids=1,2,3,4

我当前的实现是执行以下操作：

# filter
class ProjectImageFilter(django_filters.FilterSet):
    """
    Filter on existing fields, or defined query_params with
    associated functions
    """
    ids = django_filters.MethodFilter(action='id_list')

    def id_list(self, queryset, value):
        """
        Filter by IDs by passing in a query param of this structure
            `?ids=265,263`
        """
        id_list = value.split(',')
        return queryset.filter(id__in=id_list)

    class Meta:
        model = ProjectImage
        fields = ['ids',]


# viewset
class Images(viewsets.ModelViewSet):
    """
    Images associated with a project
    """
    serializer_class = ImageSerializer
    queryset = ProjectImage.objects.all()
    filter_class = ProjectImageFilter

---

但是，在这种情况下，

projectmagefilter

需要指定一个模型（

projectmage

）。有没有一种方法可以让我一般地定义这个过滤器，这样我就可以在不同模型的多个视图集上使用它？

一个没有

django过滤器的解决方案是只
super（）
覆盖
获取查询集
。以下是一个例子：

class MyViewSet(view.ViewSet):

    # your code

    def get_queryset(self):
        queryset = super(MyViewSet, self).get_queryset()

        ids = self.request.query_params.get('ids', None)
        if ids:
            ids_list = ids.split(',')
            queryset = queryset.filter(id__in=ids_list)

        return queryset

这不是一个解决方案，但是出于同样的原因，我在模型中添加了额外的字段来存储文件名甚至文件关键字。这样我就可以轻松地按名称和关键字搜索我的文件。