django rest_framework 自定义403 Forbidden错误页面

django本来有是可以很方便自定义HTTP错误页面的,网上资料一大把。核心是在项目的urls代码中增加handler403的定义,比如:

handler403 = "app.views.your_custom_view"

404,500都是一样的,重新定义handler404和handler500就行了。

但是我发现加上rest_framework之后,handler404和handler500都没问题,就是handler403怎么都不行。后来看了rest_framework的文档,才明白,rest_framework文档写的很明白,它会拦截PermissionDenied异常,所以自定义的handler403是不会起作用的。

我后来找到一种方法,自己写一个middleware,拦截403错误。

  1. 写一个自定义的的middlewares.py
复制代码
from django.shortcuts import render
from django.utils.deprecation import MiddlewareMixin

class CustomExceptionMiddleware(MiddlewareMixin):
    def process_response(self, request, response):
        if response.status_code == 500:
            error_message = f"error message: 500 server internal error"
            return render(request, 'error.html', {'error_message': error_message})
        elif response.status_code == 404:
            error_message = f"error message: 404 server internal error"
            return render(request, 'error.html', {'error_message': error_message})
        elif response.status_code == 403:
            error_message = f"error message: 403 server internal error"
            return render(request, 'error.html', {'error_message': error_message})
        else:
            return response

这里面对response进行拦截,如果status code是403则返回指定页面,否则返回原来的response

  1. 修改项目的settings,把刚写的middleware加上去:
复制代码
MIDDLEWARE = [

    'bid_request_system_app.commons.middlewares.CustomExceptionMiddleware'
]

就可以了