Django-rest-framework：从 .dispatch 返回响应

Question

我正在基于django-rest-framework 的应用程序中进行（或多或少）自定义身份验证，我只需要调用另一个微服务来询问令牌是否有效以及与之关联的用户名/用户ID . （我没有供用户使用的本地表）。

没有找到开箱即用的解决方案，我将覆盖 dispatch 方法（我使用基于 APIView 的视图），我在其中向远程服务发出请求，如果响应不是200，我要返回403错误。

这是我的代码：

def dispatch(self, request, *args, **kwargs):
    try:
        user_info = user_from_token_in_request(request)
        return super().dispatch(*args, **kwargs)
    except:
        return Response(
            "No Auth Token provided or bad Auth Token"
            status.HTTP_403_FORBIDDEN,
        )

但是，当我传递一个无效的令牌时，我得到了这个错误：AssertionError: .accepted_renderer not set on Response，因为在处理dispatch 方法时响应上下文没有被初始化。

有没有一种更合适的方法？

Answer 1

为了回答最初的问题，handle_exception 在dispatch 内部被调用，因此如果您覆盖dispatch，它不会自动为您调用。但是你可以直接调用它：

from rest_framework.exceptions import PermissionDenied

class SometimesErrorMixin:
    should_error = None

    def dispatch(self, request, *args, **kwargs):

        if not self.should_error:
            return super().dispatch(request, *args, **kwargs)

        # Lifted from DRF
        self.args = args
        self.kwargs = kwargs
        request = self.initialize_request(request, *args, **kwargs)
        self.request = request
        self.headers = self.default_response_headers  # deprecate?

        response = self.handle_exception(PermissionDenied())
        self.response = self.finalize_response(request, response, *args, **kwargs)
        return self.response

将此 Mixin 放在 ViewSet 的最左侧以使用该行为。显然你可以用适当的逻辑替换should_error。

Answer 2

尝试引发DRF 提供的异常之一，而不是返回Response，它的异常处理程序将自动处理其余部分。

from rest_framework.exceptions import PermissionDenied

def dispatch(self, request, *args, **kwargs):
    try:
        # ...
    except:
        raise PermissionDenied("NO Auth Token provided")

话虽如此，不建议在视图本身中编写身份验证代码。您应该在自定义身份验证/权限类中完成此操作。

from rest_framework.authentication import BaseAuthentication

class MyAuthentication(BaseAuthentication):
    def authenticate(self, request):
        try:
            user_info = user_from_token_in_request(request)
        except:
            raise AuthenticationFailed('No auth token provided')

class MyApiView(APIView):
    authentication_classes = [MyAuthentication]

还要确保您已在其余框架设置中定义 EXCEPTION_HANDLER。

REST_FRAMEWORK = {
    'EXCEPTION_HANDLER': 'rest_framework.views.exception_handler'
}