Django中如何在自定义视图内使用ModelViewSet视图的GET返回数据

现有实现方式（直接调用视图GET逻辑） #

你可以直接手动构造请求调用ObjectView的GET对应方法（列表对应list、单条详情对应retrieve），完全复用原有视图的所有逻辑（包含权限校验、分页、过滤、排序等你已经配置的能力），示例代码如下：

from rest_framework.request import Request
from rest_framework.test import APIRequestFactory

# 在自定义视图的业务逻辑中使用
def your_custom_logic(request):
    # 构造请求对象，继承原有请求的认证、用户信息避免权限校验失败
    factory = APIRequestFactory()
    raw_request = factory.get('/your-object-endpoint/')
    raw_request.user = request.user
    raw_request.auth = request.auth
    drf_request = Request(raw_request)

    # 调用列表查询GET逻辑
    view = ObjectView.as_view(actions={'get': 'list'})
    list_response = view(drf_request)
    list_data = list_response.data # 拿到和接口返回完全一致的列表数据

    # 调用单条查询GET逻辑
    view = ObjectView.as_view(actions={'get': 'retrieve'})
    detail_response = view(drf_request, pk=目标对象ID)
    detail_data = detail_response.data # 拿到单条对象数据

该方案的缺点是耦合度高、构造请求流程繁琐、性能略低，只适合临时快速复用的场景。

更优解耦方案（推荐） #

你需要复用的核心能力是数据查询规则+序列化规则，完全可以把这部分逻辑抽离出来，不需要绕视图层调用，维护性和性能都更好，分为两种场景处理：

场景1：不需要原有视图的分页、过滤、权限能力 #

直接复用序列化器即可，示例代码：

# 获取列表序列化数据
queryset = ObjectModel.objects.all()
# 可以自行添加过滤条件：queryset = queryset.filter(自定义过滤规则)
serializer = ObjectSerializer(queryset, many=True)
list_data = serializer.data

# 获取单条序列化数据
obj = ObjectModel.objects.get(pk=目标对象ID)
serializer = ObjectSerializer(obj)
detail_data = serializer.data

场景2：需要保留原有视图的分页、过滤、权限能力 #

把公共逻辑抽成公共方法，两边视图都调用即可，示例代码：

1. 先改造ObjectView抽离公共逻辑：

class ObjectView(viewsets.ModelViewSet):
    # 注意你原有代码这里存在笔误：ObjectSerialzer 少了字母i，修正为ObjectSerializer
    serializer_class = serializers.ObjectSerializer
    queryset = ObjectModel.objects.all()

    # 抽离列表数据处理公共方法
    def get_processed_list_data(self, request):
        queryset = self.filter_queryset(self.get_queryset())
        page = self.paginate_queryset(queryset)
        if page is not None:
            serializer = self.get_serializer(page, many=True)
            return self.get_paginated_response(serializer.data).data
        serializer = self.get_serializer(queryset, many=True)
        return serializer.data
    
    # 原有的list接口逻辑复用抽离的方法
    def list(self, request, *args, **kwargs):
        data = self.get_processed_list_data(request)
        return Response(data)

2. 在自定义视图中直接调用抽离的公共方法：

class YourCustomView(APIView):
    def get(self, request, *args, **kwargs):
        object_view = ObjectView()
        # 直接拿到处理后的列表数据，包含原有视图的过滤、分页能力
        object_list_data = object_view.get_processed_list_data(request)
        # 后续你的自定义业务逻辑
        return Response({
            "custom_data": "你的其他业务返回",
            "object_list": object_list_data
        })

内容的提问来源于stack exchange，提问作者ag2byte