django怎么自己创建一个中间件

• 中间件是什么？

中间件是类似flask函数中钩子函数的东西。可以在请求视图函数前，或者视图函数响应后处理某些事情。中间件对全部视图都有效！

中间件一般会有两个方法，process_request和process_response。分别用来处理请求到达视图函数前，到达视图函数后的事情。

 

• 编写一个自定义的中间件

我们可以根据业务需要自己编写一个中间件。在编写中间件前，你需要读一下django.utils.deprecation.MiddlewareMixin的源码。

class MiddlewareMixin(object):
    def __init__(self, get_response=None):
        self.get_response = get_response
        super(MiddlewareMixin, self).__init__()

    def __call__(self, request):
        response = None
        if hasattr(self, 'process_request'):
            response = self.process_request(request)
        if not response:
            response = self.get_response(request)
        if hasattr(self, 'process_response'):
            response = self.process_response(request, response)
        return response

方法名	作用	返回值	备注
__init__(self, get_response=None)	服务器启动(重启)时执行一次	无	--
process_request(self, request)	在视图执行之前调用	返回 `None`: 会执行视图 返回 `HttpResponse`: 不会再执行视图	--
process_response(self, request, response)	在视图执行完之后调用	必须返回HttpResponse对象	--

 

编写自定义的中间件：

class My_middleware(MiddlewareMixin):
    def __init__(self,get_response=None):
        super().__init__(get_response)

    def process_request(self,request):
        """
        视图函数前执行的，一旦这里有return将到达不了视图函数
        request是浏览器发过来的请求
        """
        print("视图函数执行前")

    def process_response(self,request,response):
        """
        视图函数后执行的
        request是浏览器发过来的请求
        response是视图函数返回的响应
        """
        print("视图函数执行后")

        # 因为你拿了视图函数的response，就算处理的业务与response无关也要返回response
        return response

 

• 中间件的注册

MIDDLEWARE = [
    ...
    'middlewares.My_middleware'    # 写上自己的路径
]

 

• 多个中间件的执行顺序

！！！如果你定义了多个中间件，记得注意一下中间件摆在setting.py文件的位置：

• 注册中间件设置是由上到下执行的，所以上面的中间件会先被注册
• 对于视图函数执行前的process_request，先注册的中间件会先执行
• 对于视图函数执行后的process_response，后注册的中间件会先执行

例如：

先注册中间件1，后注册中间件2。
输出结果是：

before 视图（中间件1）
before 视图 2（中间件2）
==index==
after 视图 2（中间件2）
after 视图（中间件1）