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错误。
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
- 修改项目的settings,把刚写的middleware加上去:
MIDDLEWARE = [
'bid_request_system_app.commons.middlewares.CustomExceptionMiddleware'
]
就可以了