[英]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.