-1

我将 django-filter 与 DRF 一起使用。我有一个favourite模型，它通过 GenericRelation 链接到其他几个模型。为了过滤具有 favourite-flag 的条目，我创建了一个 custom FavouriteFilter，并将其添加到相应的模型中。我想查询content_type_id相应模型的 ，以限制Favourite. 但是，我不知道如何model将FavouriteFilter.

这是一个代码片段来说明这个问题：

class ProjectFilter(BaseFilter):

    favourite_only = FavouriteFilter()


class FavouriteFilter(django_filters.BooleanFilter):
    """
    A custom filter which returns a users favourites of an element
    """

    def __init__(self, *args, **kwargs):
        # gettext_lazy breaks the OpenAPI generation => use gettext instead
        kwargs['label'] = gettext("My favourites")
        super(FavouriteFilter, self).__init__(*args, **kwargs)

    def filter(self, qs, value):
        if value == True:
            user = get_current_user()
            content_type = ContentType.objects.get_for_model(<model>)
            return qs.filter(pk__in=Favourite.objects
                             .filter(owner_id=user)
                             .filter(content_type_id=content_type)
                             .values_list('object_id', flat=True)
                             )
        else:
            return qs

在此示例中，<model>缺少 - 属性。如何将此信息传递Project给过滤器？

4

1 回答 1

0

关键字参数可以传递给过滤器，但需要在调用 super() 方法之前从 kwarg-dict 中删除。否则它们会被传递给超类，超类的__init__()-method 不知道关键字并且 aTypeError被抛出：

TypeError: __init__() got an unexpected keyword argument 'model'

在上面的例子中，超类django_filters.BooleanFilter分别是django_filters.Filter.

使用dict.pop()- 方法，关键字从 kwargs-dictionary 中删除，同时我们可以保存它以供进一步使用。由于content_type初始化后永远不会改变，因此已经可以在__init__().

这是上面代码的一个工作示例，Project我想传递给过滤器的 django-model 在哪里：

class ProjectFilter(BaseFilter):

    favourite_only = FavouriteFilter(model=Project)


class FavouriteFilter(django_filters.BooleanFilter):
    """
    A custom filter which returns a users favourites of an element
    """

    def __init__(self, *args, **kwargs):
        # gettext_lazy breaks the OpenAPI generation => use gettext instead
        kwargs['label'] = gettext("My favourites")
        model = kwargs.pop('model')
        self.content_type = ContentType.objects.get_for_model(model)
        super(FavouriteFilter, self).__init__(*args, **kwargs)

    def filter(self, qs, value):
        if value == True:
            user = get_current_user()
            return qs.filter(pk__in=Favourite.objects
                             .filter(owner_id=user)
                             .filter(content_type_id=self.content_type)
                             .values_list('object_id', flat=True)
                             )
        else:
            return qs

对于我正在寻找使用过滤器的模型的特定用例，该模型可通过查询集作为qs.model. 代码片段如下所示：

class ProjectFilter(BaseFilter):

    favourite_only = FavouriteFilter()


class FavouriteFilter(django_filters.BooleanFilter):
    """
    A custom filter which returns a users favourites of an element
    """

    def __init__(self, *args, **kwargs):
        # gettext_lazy breaks the OpenAPI generation => use gettext instead
        kwargs['label'] = gettext("My favourites")
        super(FavouriteFilter, self).__init__(*args, **kwargs)

    def filter(self, qs, value):
        if value == True:
            user = get_current_user()
            content_type = ContentType.objects.get_for_model(qs.model)
            return qs.filter(pk__in=Favourite.objects
                             .filter(owner_id=user)
                             .filter(content_type_id=content_type)
                             .values_list('object_id', flat=True)
                             )
        else:
            return qs
于 2020-09-18T08:15:37.193 回答