Django Rest Framework - 基于标题数据针对单个 URL 注册多个视图集

如何解决Django Rest Framework - 基于标题数据针对单个 URL 注册多个视图集

我需要使用多个具有相同 URL 的 ViewSet，其中需要根据请求 header 逻辑动态选择活动的 ViewSet。 Django rest 框架允许您这样做以针对不同的 url 注册视图集：

router.register(r"type_1",Type1ViewSet,basename="type_1")
router.register(r"type_2",Type2ViewSet,basename="type_2")

但是，就我而言，视图集非常相似。所以我想在 urls.py 文件中执行此操作：

if request.header['flag'] is True:
    router.register(r"type",basename="type_1")
else: 
    router.register(r"type",basename="type_2")

就我而言，以下方法不起作用：

• 使用单个 ViewSet 但从标头逻辑中选择不同的 Serializers，而不是处理多个 ViewSet。

有没有办法访问 request 中的 urls.py 对象，以便我可以使用它来编排我的条件？如果没有，如何实现？

解决方法

你不能实现你想要的，对同一条路线使用两个不同的视图是不可能的。

但是，如果您的两个视图集之间唯一改变的是序列化程序，则只需使用方法 get_serializer_class。

class YourViewset(viewsets.ModelViewSet):
    # Your attributes ....

    def get_serializer_class(self):
        if self.request.header['flag'] is True: # your condition
            return Type1Serializer
         return Type2Serializer