django校验token自定义中间件

from django.utils.deprecation import MiddlewareMixin
from django.http import JsonResponse
 
class TokenMiddleware(MiddlewareMixin):
    def process_request(self, request):
        # 获取请求头中的token
        token = request.META.get('HTTP_TOKEN', None)
        # 验证token是否正确，这里仅作为示例，实际应该与你的身份验证系统对接
        if token != 'your-custom-token':
            return JsonResponse({'error': 'Invalid token'}, status=401)
 
    def process_response(self, request, response):
        # 如果需要在响应中添加额外的处理，可以在此方法中实现
        return response

这段代码定义了一个名为TokenMiddleware的Django中间件，用于验证HTTP请求中的token是否正确。如果token不正确，则中间件会中断请求并返回一个包含错误信息的JSON响应，状态码为401。这个例子展示了如何在Django项目中实现自定义的中间件来处理请求验证。