如何在不调用 API 的情况下从 DRF 中的 ModelViewSet 获取数据

Question

我将把我所有的 API 转换为gRPC调用。 目前我能够将所有ViewSet转移到gRPC 中（示例代码添加到这个问题的末尾）。 但是ModelViewSet ，它会出现这样的错误。

Traceback (most recent call last):
  File "/home/wasdkiller/PycharmProjects/ocsa/dataengine-service/venv/lib/python3.6/site-packages/grpc/_server.py", line 435, in _call_behavior
    response_or_iterator = behavior(argument, context)
  File "/home/wasdkiller/PycharmProjects/ocsa/dataengine-service/servicers/tenant/main.py", line 15, in get_tenant
    data = ClientViewSet().list(request=original_request, )
  File "/home/wasdkiller/PycharmProjects/ocsa/dataengine-service/common/lib/decorators.py", line 128, in wrapper_format_response
    final_data = call_func(func, self, request, transaction, exception, *args, **kwargs)
  File "/home/wasdkiller/PycharmProjects/ocsa/dataengine-service/common/lib/decorators.py", line 99, in call_func
    return func(self, request, *args, **kwargs)
  File "/home/wasdkiller/PycharmProjects/ocsa/dataengine-service/api_v1/viewsets.py", line 471, in list
    data = super().list(request, *args, **kwargs).data
  File "/home/wasdkiller/PycharmProjects/ocsa/dataengine-service/venv/lib/python3.6/site-packages/rest_framework/mixins.py", line 38, in list
    queryset = self.filter_queryset(self.get_queryset())
  File "/home/wasdkiller/PycharmProjects/ocsa/dataengine-service/venv/lib/python3.6/site-packages/rest_framework/generics.py", line 158, in filter_queryset
    queryset = backend().filter_queryset(self.request, queryset, self)
AttributeError: 'ClientViewSet' object has no attribute 'request'

所以我的viewsets.py看起来像这样（ format_response装饰器将其转换为Response对象）

class ClientViewSet(viewsets.ModelViewSet):
    queryset = Client.objects.all()
    serializer_class = ser.ClientSerializer

    @format_response(exception=False)
    def list(self, request, *args, **kwargs):
        data = super().list(request, *args, **kwargs).data
        # data = {}
        return data, True, HTTPStatus.OK, 'data retrieve successfully'

当我将其称为 API 时，它可以完美运行。 但我想在不调用 API 的情况下做同样的事情。 这是我如何解决的，

from django.http import HttpRequest
from rest_framework.request import Request


# creating request object
django_request = HttpRequest()
django_request.method = 'GET'
drf_request = Request(django_request)

data = ClientViewSet().list(request=drf_request)
print(f'data: {data.data}')

ClientViewSet 中的super() ClientViewSet的问题，但是如果我取消注释data = {}并注释掉调用super() function，它适用于 ZDB974238714CA81434A7CE1D03 和上述方法。 I go through inside the DRF code base where error occurred, when calling the API the self object has the request object and above method it doesn't exist.

Answer 1

简而言之，您需要在视图集上调用as_view()和 map http 动词，然后使用适当的 HttpRequest 调用 function。

django 中的所有视图最终都是函数，DRF 是最重要的糖。 路由完成后，“ViewSet”以正常方式不存在，作为独立的 class。

django_request = HttpRequest()
django_request.method = 'GET'

my_view = ClientViewSet.as_view({'get': 'list', 'post':'create'})
data = my_view(request=django_request)
print(f'data: {data.data}')

如果您想要详细路线（ users/37 ，...），那么您需要创建一个新视图 function 映射到这些视图集函数。 这不可能是同一个视图 function 因为get现在需要在视图集中指向不同的 function 而不是列表案例。 查看routers.py 源代码了解发生了什么以及映射到哪里。

# map http 'get' to the 'retrive' function on the viewset
my_view = ClientViewSet.as_view({'get': 'retrieve', ...})

# pass in the kwarg the URL routing would normally extract
client_pk = 9329032 
data = my_view(request=django_request, pk=client_pk)

如果您想查看视图集的所有映射是什么，那么您可以使用以下代码段将它们打印出来：

router = SimpleRouter()
router.register("client", ClientViewSet, basename="client")
for url in router.urls:  # type: URLPattern
    print(f"{url.pattern} ==> {url.callback}")
    for verb, action in url.callback.actions.items():
        print(f"    {verb} -> {action}")

# output will be something like this
^client/$ ==> <function ClientViewSet at 0x11b91c280>
    get -> list
^client/(?P<pk>[^/.]+)/$ ==> <function ClientViewSet at 0x11b91c3a0>
    get -> retrieve
    put -> update
    patch -> partial_update
    delete -> destroy

你传递给这个视图的kwargs将取决于你的 ViewSet 中的设置，比如lookup_url_kwarg ，但在大多数情况下它们会很简单。