rest_framework_django学习笔记三(异常、登录认证)

from rest_framework.views import exception_handler as drf_exception_handler
from rest_framework.response import Response
from rest_framework import status
from django.contrib.auth import authenticate
from django.utils.translation import ugettext_lazy as _
 
def custom_exception_handler(exc, context):
    response = drf_exception_handler(exc, context)
    if response is None:
        # 未被DRF处理的异常
        return Response({'detail': str(exc)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
    return response
 
def custom_authenticate(request):
    # 示例自定义认证逻辑
    # 假设我们使用一个自定义的认证方式，比如token
    auth_header = request.META.get('HTTP_AUTHORIZATION', b'')
    token = auth_header.decode()
 
    user = authenticate(request, token=token)
    if user is not None:
        # 用户认证成功
        request.user = user
        return True
    else:
        # 用户认证失败
        return False
 
# 在settings.py中设置
REST_FRAMEWORK = {
    'EXCEPTION_HANDLER': 'path.to.custom_exception_handler',  # 指向自定义异常处理函数的路径
    'DEFAULT_AUTHENTICATION_CLASSES': (
        'path.to.custom_authenticate',  # 指向自定义认证函数的路径
    ),
}

在这个例子中，我们定义了一个自定义异常处理函数custom_exception_handler，它会被用来替换默认的异常处理函数。同时，我们定义了一个自定义认证方法custom_authenticate，它可以用来替换默认的认证方式。在settings.py文件中，我们通过指定REST_FRAMEWORK字典来使用这些自定义设置。这样，我们就可以根据自己的需求来扩展和自定义DRF的行为。