繁体   English   中英

DRF - 创建一个 Viewset Mixin/Baseclass 来继承一个 viewset-action

[英]DRF - Create a Viewset Mixin/Baseclass to inherit a viewset-action

我有一个网站,它本质上是我参与的 DnD 活动的 wiki。因此,它有关于生物、角色、位置等的文章。 我想使用 Viewsets 轻松访问它们,并希望使用 Viewset 操作(与自定义路由器一起)能够不通过 pk 而是通过各种查询参数查找单个记录。

我已经有了一些适用于此的东西,现在我想对其应用一些继承,以免重复自己。 我想做的是这样的:

class WikiBaseViewset (viewsets.ModelViewSet):
    detail_with_params_url_pattern_suffix: str

    @action(detail=True, url_name="detail-params", url_path=detail_with_params_url_pattern_suffix)
    def detail_through_params(self, request, **kwargs):
        if self.detail_with_params_url_pattern_suffix == "":
            raise InvalidViewsetURLException("URL of view 'detail_through_params' of WikiBaseViewset is not defined!")

        model = self.serializer_class.Meta.model
        instance = get_object_or_404(model, **kwargs)
        serializer = self.get_serializer(instance)
        return Response(serializer.data)


class CharacterSerializer (serializers.HyperlinkedModelSerializer):
    class Meta:
        model = wiki_models.Character
        fields = '__all__'


class CharacterViewSet(WikiBaseViewset):
    """Called with URLs: character, character/<str: name>"""
    serializer_class = CharacterSerializer
    queryset = wiki_models.Character.objects.all()
    detail_with_params_url_pattern_suffix = "(?P<name__iexact>.+)"

但是,我正在为装饰器绝对需要基类中的 URL 参数这一事实而苦苦挣扎。 否则,由于 NameError 抱怨detail_with_params_url_pattern_suffix ,代码无法编译。 如果您要在基类中设置detail_with_params_url_pattern_suffix=""以便在编译代码时不会出现错误,那仍然无关紧要,因为到目前为止我的实验中的装饰器仍然从WikiBaseViewset不是CharacterViewSet

我如何重写我的 BaseClass 以使其有效? 甚至有办法吗?

对于这个问题,我没有找到一个完全令人满意的答案,但最终还是默认了这个解决方案,因为它比复制粘贴要好。

您可能无法继承视图集操作,但您肯定可以继承单个方法,然后只需在子级中覆盖它们并在顶部放置一个装饰器。 这导致了这种结构:

class WikiBaseViewset (viewsets.ModelViewSet):
    detail_with_params_url_pattern_suffix: str

    def detail_through_params(self, request, **kwargs):
        model = self.serializer_class.Meta.model
        instance = get_object_or_404(model, **kwargs)
        serializer = self.get_serializer(instance)
        return Response(serializer.data)


class CharacterSerializer (serializers.HyperlinkedModelSerializer):
    class Meta:
        model = wiki_models.Character
        fields = '__all__'


class CharacterViewSet(WikiBaseViewset):
    """Called with URLs: character, character/<str: name>"""
    serializer_class = CharacterSerializer
    queryset = wiki_models.Character.objects.all()

    @action(detail=True, url_name="detail-params", url_path="(?P<name__iexact>.+)")
    def detail_through_params(self, request, **kwargs):
        return super().detail_through_params(request, **kwargs)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM