Django之中间件

前戏

我们在前面的课程中已经学会了给视图函数加装饰器来判断是用户是否登录,把没有登录的用户请求跳转到登录页面。我们通过给几个特定视图函数加装饰器实现了这个需求。但是以后添加的视图函数可能也需要加上装饰器,这样是不是稍微有点繁琐。

学完今天的内容之后呢,我们就可以用更适宜的方式来实现类似给所有请求都做相同操作的功能了。

中间件

中间件介绍

什么是中间件?

官方的说法:中间件是一个用来处理Django的请求和响应的框架级别的钩子。它是一个轻量、低级别的插件系统,用于在全局范围内改变Django的输入和输出。每个中间件组件都负责做一些特定的功能。

但是由于其影响的是全局,所以需要谨慎使用,使用不当会影响性能。

说的直白一点中间件是帮助我们在视图函数执行之前和执行之后都可以做一些额外的操作,它本质上就是一个自定义类,类中定义了几个方法,Django框架会在处理请求的特定的时间去执行这些方法。

我们一直都在使用中间件,只是没有注意到而已,打开Django项目的Settings.py文件,看到下图的MIDDLEWARE配置项。

MIDDLEWARE = [
    'django.middleware.security.SecurityMiddleware',
    'django.contrib.sessions.middleware.SessionMiddleware',
    'django.middleware.common.CommonMiddleware',
    'django.middleware.csrf.CsrfViewMiddleware',
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    'django.contrib.messages.middleware.MessageMiddleware',
    'django.middleware.clickjacking.XFrameOptionsMiddleware',
]

MIDDLEWARE配置项是一个列表,列表中是一个个字符串,这些字符串其实是一个个类,也就是一个个中间件。

我们之前已经接触过一个csrf相关的中间件了?我们一开始让大家把他注释掉,再提交post请求的时候,就不会被forbidden了,后来学会使用csrf_token之后就不再注释这个中间件了。

那接下来就学习中间件中的方法以及这些方法什么时候被执行。

自定义中间件

中间件可以定义五个方法,分别是:(主要的是process_request和process_response)

  • process_request(self,request)
  • process_view(self, request, view_func, view_args, view_kwargs)
  • process_template_response(self,request,response)
  • process_exception(self, request, exception)
  • process_response(self, request, response)

以上方法都有  参数、 执行时间 、执行顺序、 返回值 

返回值可以是None:如果是None,则继续按照django定义的规则向后继续执行

或一个HttpResponse对象,如果是HttpResponse对象,则直接将该对象返回给用户。

process_request

HttpResponse对象
当前中间件后面的中间件的process_request和process_response方法、视图函数都不执行
执行当前中间的process_response方法以及之前的中间的process_response方法

我们来看看多个中间件时,Django是如何执行其中的process_request方法的。

class MD1(MiddlewareMixin):

def process_request(self, request):
request.test = 'test'
print("MD1里面的 process_request",'中间件的request:',type(request))
# return HttpResponse('MD1')


class MD2(MiddlewareMixin):
def process_request(self, request):
print("MD2里面的 process_request",'中间件的request:',type(request))
# return HttpResponse('MD2')

在views中定义index视图

from django.shortcuts import render,HttpResponse,redirect


def index(request):
    print('这里是 index 视图函数','request类型:',type(request))
    print('这里是 index 视图函数','request类型:',type(request))
return HttpResponse('o98k')

填好对应的urls后

在settings.py的MIDDLEWARE配置项中注册上述两个自定义中间件:

MIDDLEWARE = [
    'django.middleware.security.SecurityMiddleware',
    'django.contrib.sessions.middleware.SessionMiddleware',
    'django.middleware.common.CommonMiddleware',
    'django.middleware.csrf.CsrfViewMiddleware',
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    'django.contrib.messages.middleware.MessageMiddleware',
    'django.middleware.clickjacking.XFrameOptionsMiddleware',
    'middlewares.MD1',  # 自定义中间件MD1
    'middlewares.MD2'  # 自定义中间件MD2
]

此时,我们访问一个视图,会发现终端中打印如下内容:

MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
test
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>

把MD1和MD2的位置调换一下,再访问一个视图,会发现终端中打印的内容如下:

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
test
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>

看结果我们知道:视图函数还是最后执行的,MD2比MD1先执行自己的process_request方法。

在打印一下两个自定义中间件中process_request方法中的request参数,会发现它们是同一个对象。

如果在MD1中的return HttpResponse('MD1')注释取消,在访问下一个试图,终端会打印

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>

 

如果换成MD2中的return HttpResponse('MD1')注释取消,在访问下一个试图,终端会打印

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'> 

看结果得到,只会执行到HttpResponse返回的方法,视图都不会执行了、

由此总结一下:

  1. 中间件的process_request方法是在执行视图函数之前执行的。
  2. 当配置多个中间件时,会按照MIDDLEWARE中的注册顺序,也就是列表的索引值,从前到后依次执行的。
  3. 不同中间件之间传递的request都是同一个对象.
  4. 在视图执行之前,可以通过中间件给request赋予属性值
  5. 如果request中有 HttpResponse对象,那就直接返给浏览器,不再执行视图方法了

1. 执行时间 
  在视图函数执行之前
2. 参数
  request ——》 视图函数中用到的request

3. 执行顺序
  按照stetings.py文件注册顺序顺序执行
4. 返回值
  1.None 正常流程走

  2.如果是HttpResponse对象,Django将不执行视图函数,而将响应对象返回给浏览器

process_response

它有两个参数,一个是request,一个是response,request就是上述例子中一样的对象,response是视图函数返回的HttpResponse对象。该方法的返回值也必须是HttpResponse对象。

给上述的M1和M2加上process_response方法:

from django.utils.deprecation import MiddlewareMixin
from django.shortcuts import HttpResponse


class MD1(MiddlewareMixin):

def process_request(self, request):
# request.test = 'test'
print("MD1里面的 process_request", '中间件的request:', type(request))
# return HttpResponse('MD1')

def process_response(self, request, response):
print("MD1里面的 process_response")
# return response


class MD2(MiddlewareMixin):
def process_request(self, request):
print("MD2里面的 process_request", '中间件的request:', type(request))
# return HttpResponse('MD2')

def process_response(self, request, response):
print("MD2里面的 process_response")
# return response

访问一个视图,看一下终端的输出:

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_response
MD2里面的 process_response
#之后报错,说是找不到'NoneType' object has no attribute 'get':其实是返回值中没有返回导致的response

把MD1中的  return response 注释取消,访问下视图,看一下终端结果

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_response
MD2里面的 process_response
#之后报错

把MD2(MD1继续注释掉)中的  return response 注释取消,访问下视图,看一下终端结果

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_response
MD2里面的 process_response
#之后报错

把MD1和MD2 中的return response 注释都取消,访问下视图,看一下终端结果

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_response
MD2里面的 process_response

  

看结果可知:

process_response方法是在视图函数之后执行的,并且顺序是MD1比MD2先执行。(此时settings.py中 MD2比MD1先注册)

多个中间件中的process_response方法是按照MIDDLEWARE中的注册顺序倒序执行的,也就是说第一个中间件的process_request方法首先执行,而它的process_response方法最后执行,最后一个中间件的process_request方法最后一个执行,它的process_response方法是最先执行。

1. 执行时间
  在视图函数执行之后
2. 参数
  request ——》 视图函数中用到的request
  response —— 》视图函数中返回的response
3. 返回值
  必须是response对象
4. 执行顺序
  按照stetings.py文件注册倒序顺序执行

process_view

process_view(self, request, view_func, view_args, view_kwargs)

该方法有四个参数

request是HttpRequest对象。

view_func是Django即将使用的视图函数。 (它是实际的函数对象,而不是函数的名称作为字符串。)

view_args是将传递给视图的位置参数的列表.

view_kwargs是将传递给视图的关键字参数的字典。 view_args和view_kwargs都不包含第一个视图参数(request)。

Django会在调用视图函数之前调用process_view方法。

它应该返回None或一个HttpResponse对象。 如果返回None,Django将继续处理这个请求,执行任何其他中间件的process_view方法,然后在执行相应的视图。 如果它返回一个HttpResponse对象,Django不会调用适当的视图函数。 它将执行中间件的process_response方法并将应用到该HttpResponse并返回结果。

 给MD1和MD2添加process_view方法:

from django.utils.deprecation import MiddlewareMixin
from django.shortcuts import HttpResponse


class MD1(MiddlewareMixin):

def process_request(self, request):
# request.test = 'test'
print("MD1里面的 process_request", '中间件的request:', type(request))
# return HttpResponse('MD1')

def process_response(self, request, response):
print("MD1里面的 process_response")
return response

def process_view(self, request, view_func, view_args, view_kwargs):
print('这是MD1中的process_view方法')
print(view_func, view_func.__name__)

class MD2(MiddlewareMixin):
def process_request(self, request):
print("MD2里面的 process_request", '中间件的request:', type(request))
# return HttpResponse('MD2')

def process_response(self, request, response):
print("MD2里面的 process_response")
return response

def process_view(self, request, view_func, view_args, view_kwargs):
print('这是MD2中的process_view方法')
print(view_func, view_func.__name__)

访问index视图函数,看一下输出结果:

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
这是MD2中的process_view方法
<function index at 0x0593BC90> index
这是MD1中的process_view方法
<function index at 0x0593BC90> index
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_response
MD2里面的 process_response

process_view方法是在process_request之后,视图函数之前执行的,执行顺序按照MIDDLEWARE中的注册顺序从前到后顺序执行的

此时urls文件修改为

url(r'^index/(?P<num>d+)', views.index),  # 网页访问后边要输入数字才可以访问

视图中修改为

from django.shortcuts import render,HttpResponse,redirect


def index(request,num):
    # print(request.test)
    print('这里是 index 视图函数','request类型:',type(request))
    print(num)  # 多了一个num并参数打印出来
    return HttpResponse('o98k')

request 没有HttpResponse对象时候,访问一个视图,后边加上111

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
这是MD2中的process_view方法
<function index at 0x059DBCD8> index
这是MD1中的process_view方法
<function index at 0x059DBCD8> index
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>
111  # 在这里得到 拿到process_view方法传来的参数
MD1里面的 process_response
MD2里面的 process_response

可以得知:

1. 执行时间
  在process_request之后,以及路由匹配之后,在视图函数执行之前
2. 参数
  view_func ——》 要执行的视图函数
  view_args ——》 视图函数的位置参数
  view_kwargs ——》 视图函数的关键字参数
3. 返回值
  None 正常走

4. 执行顺序
  按照stetings.py文件注册顺序顺序执行

process_exception

process_exception(self, request, exception)

该方法两个参数:

一个HttpRequest对象

一个exception是视图函数异常产生的Exception对象。

这个方法只有在视图函数中出现异常了才执行,它返回的值可以是一个None也可以是一个HttpResponse对象。如果是HttpResponse对象,Django将调用模板和中间件中的process_response方法,并返回给浏览器,否则将默认处理异常。如果返回一个None,则交给下一个中间件的process_exception方法来处理异常。它的执行顺序也是按照中间件注册顺序的倒序执行。

 给MD1和MD2添加上这个方法:

from django.utils.deprecation import MiddlewareMixin


class MD1(MiddlewareMixin):

    def process_request(self, request):
        print("MD1里面的 process_request")

    def process_response(self, request, response):
        print("MD1里面的 process_response")
        return response

    def process_view(self, request, view_func, view_args, view_kwargs):
        print("-" * 80)
        print("MD1 中的process_view")
        print(view_func, view_func.__name__)

    def process_exception(self, request, exception):
        print(exception)
        print("MD1 中的process_exception")


class MD2(MiddlewareMixin):
    def process_request(self, request):
        print("MD2里面的 process_request")

    def process_response(self, request, response):
        print("MD2里面的 process_response")
        return response

    def process_view(self, request, view_func, view_args, view_kwargs):
        print("-" * 80)
        print("MD2 中的process_view")
        print(view_func, view_func.__name__)

    def process_exception(self, request, exception):
        print(exception)
        print("MD2 中的process_exception")

如果视图函数中无异常,process_exception方法不执行。

想办法,在视图函数中抛出一个异常:

from django.shortcuts import render,HttpResponse,redirect


def index(request,num,):
# print(request.test)
print('这里是 index 视图函数','request类型:',type(request))
print(num)
int('aaaa') # 弄个错误
return HttpResponse('o98k')

在MD1的process_exception中返回一个响应对象:

from django.utils.deprecation import MiddlewareMixin
from django.shortcuts import HttpResponse


class MD1(MiddlewareMixin):

def process_request(self, request):
# request.test = 'test'
print("MD1里面的 process_request", '中间件的request:', type(request))
# return HttpResponse('MD1')

def process_response(self, request, response):
print("MD1里面的 process_response")
return response

def process_view(self, request, view_func, view_args, view_kwargs):
print('这是MD1中的process_view方法')
# print(view_func, view_func.__name__)

def process_exception(self,request,exception):
print(exception)
print("MD1 中的process_exception")
# return HttpResponse(str(exception)) # 返回一个响应对象


class MD2(MiddlewareMixin):
def process_request(self, request):
print("MD2里面的 process_request", '中间件的request:', type(request))
# return HttpResponse('MD2')

def process_response(self, request, response):
print("MD2里面的 process_response")
return response

def process_view(self, request, view_func, view_args, view_kwargs):
print('这是MD2中的process_view方法')
# print(view_func, view_func.__name__)

def process_exception(self,request,exception):
print(exception)
print("MD2 中的process_exception")

看输出结果:

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
这是MD2中的process_view方法
这是MD1中的process_view方法
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>
111
invalid literal for int() with base 10: 'aaaa'
MD1 中的process_exception
MD1里面的 process_response
MD2里面的 process_response

1. 执行时间(触发条件:有异常才执行)
  在视图函数之后,在process_response之前
2. 参数
  exception ——》 错误信息对象
3. 返回值
  None 正常走
  HttpResponse对象
    注册顺序之前的中间件的process_exception方法不走了。
    执行所有中间件的process_response方法

4. 执行顺序
  按照注册顺序倒叙执行

process_template_response(用的比较少)

process_template_response(self, request, response)

它的参数,一个HttpRequest对象,response是TemplateResponse对象(由视图函数或者中间件产生)。

process_template_response是在视图函数执行完成后立即执行,但是它有一个前提条件,那就是视图函数返回的对象有一个render()方法(或者表明该对象是一个TemplateResponse对象或等价方法)。

from django.utils.deprecation import MiddlewareMixin
from django.shortcuts import HttpResponse


class MD1(MiddlewareMixin):

def process_request(self, request):
print("MD1里面的 process_request", '中间件的request:', type(request))

def process_response(self, request, response):
print("MD1里面的 process_response")
return response

def process_view(self, request, view_func, view_args, view_kwargs):
print('这是MD1中的process_view方法')


def process_exception(self,request,exception):
print(exception)
print("MD1 中的process_exception")
return HttpResponse(str(exception)) # 返回一个响应对象


def process_template_response(self,request,response):
print('这是MD1中的process_template_response方法')

return response

class MD2(MiddlewareMixin):
def process_request(self, request):
print("MD2里面的 process_request", '中间件的request:', type(request))

def process_response(self, request, response):
print("MD2里面的 process_response")
return response

def process_view(self, request, view_func, view_args, view_kwargs):
print('这是MD2中的process_view方法')

def process_exception(self,request,exception):
print(exception)
print("MD2 中的process_exception")

def process_template_response(self,request,response):
print('这是MD2中的process_template_response方法')
return response

views.py中:

from django.shortcuts import render, HttpResponse, redirect


def index(request, num, ):
print('这里是 index 视图函数', 'request类型:', type(request))
ret = HttpResponse('ok')

def xxxx():
print('这是index中的xxxx')
return HttpResponse('这是index中的xxxx')

ret.render = xxxx
return ret
 

访问index视图,终端输出的结果:

MD2里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
MD1里面的 process_request 中间件的request: <class 'django.core.handlers.wsgi.WSGIRequest'>
这是MD2中的process_view方法
这是MD1中的process_view方法
这里是 index 视图函数 request类型: <class 'django.core.handlers.wsgi.WSGIRequest'>
这是MD1中的process_template_response方法
这是MD2中的process_template_response方法
这是index中的xxxx
MD1里面的 process_response
MD2里面的 process_response

从结果看出:

视图函数执行完之后,立即执行了中间件的process_template_response方法,顺序是倒序,先执行MD1的,在执行MD2的,接着执行了视图函数返回的HttpResponse对象的render方法,返回了一个新的HttpResponse对象,接着执行中间件的process_response方法。

1. 执行时间(触发条件:response对象要求有一个render方法)
  在视图函数之后,在process_response之前
2. 参数

3. 返回值
  返回response
4. 执行顺序
  按照settings.py文件中间件注册顺序倒叙执行

5.render

  render是在process_template_response执行后,process_response前执行一次

中间件的执行流程

上一部分,我们了解了中间件中的5个方法,它们的参数、返回值以及什么时候执行,现在总结一下中间件的执行流程。

请求到达中间件之后,先按照正序执行每个注册中间件的process_reques方法,process_request方法返回的值是None,就依次执行,如果返回的值是HttpResponse对象,不再执行后面的process_request方法,而是执行当前对应中间件的process_response方法,将HttpResponse对象返回给浏览器。也就是说:如果MIDDLEWARE中注册了6个中间件,执行过程中,第3个中间件返回了一个HttpResponse对象,那么第4,5,6中间件的process_request和process_response方法都不执行,顺序执行3,2,1中间件的process_response方法。

 

process_request方法都执行完后,匹配路由,找到要执行的视图函数,先不执行视图函数,先执行中间件中的process_view方法,process_view方法返回None,继续按顺序执行,所有process_view方法执行完后执行视图函数。假如中间件3 的process_view方法返回了HttpResponse对象,则4,5,6的process_view以及视图函数都不执行,直接从最后一个中间件,也就是中间件6的process_response方法开始倒序执行。

process_template_response和process_exception两个方法的触发是有条件的,执行顺序也是倒序。总结所有的执行流程如下:

 

中间件版登录验证 

中间件版的登录验证需要依靠session,所以数据库中要有django_session表。

urls.py

from django.conf.urls import url
from django.contrib import admin
from app01 import views

urlpatterns = [
    url(r'^admin/', admin.site.urls),
    url(r'^login/$', views.login, name='login'),
    url(r'^index/$', views.index, name='index'),
    url(r'^home/$', views.home, name='home'),
]

views.py

from django.shortcuts import render, HttpResponse, redirect


def index(request):
    return HttpResponse('this is index')


def home(request):
    return HttpResponse('this is home')


def login(request):
    if request.method == "POST":
        user = request.POST.get("user")
        pwd = request.POST.get("pwd")

        if user == "alex" and pwd == "alex3714":
            # 设置session
            request.session["user"] = user
            # 获取跳到登陆页面之前的URL
            next_url = request.GET.get("next")
            # 如果有,就跳转回登陆之前的URL
            if next_url:
                return redirect(next_url)
            # 否则默认跳转到index页面
            else:
                return redirect("/index/")
    return render(request, "login.html")

views.py
views.py

login.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta http-equiv="x-ua-compatible" content="IE=edge">
    <meta name="viewport" content="width=device-width, initial-scale=1">
    <title>登录页面</title>
</head>
<body>
<form action="{% url 'login' %}" method="post">
    {% csrf_token %}
    <p>
        <label for="user">用户名:</label>
        <input type="text" name="user" id="user">
    </p>
    <p>
        <label for="pwd">密 码:</label>
        <input type="text" name="pwd" id="pwd">
    </p>
    <input type="submit" value="登录">
</form>
</body>
</html>

login.html

middlewares.py

from django.utils.deprecation import MiddlewareMixin


class AuthMD(MiddlewareMixin):
    white_list = ['/login/', ]  # 白名单
    black_list = ['/black/', ]  # 黑名单

    def process_request(self, request):
        from django.shortcuts import redirect, HttpResponse

        next_url = request.path_info
        print(request.path_info, request.get_full_path())
        # 黑名单的网址限制访问
        if next_url in self.black_list:
            return HttpResponse('This is an illegal URL')
        # 白名单的网址或者登陆用户不做限制
        elif next_url in self.white_list or request.session.get("user"):
            return
        else:
            return redirect("/login/?next={}".format(next_url))

中间件
中间件

在settings.py中注册

MIDDLEWARE = [
    'django.middleware.security.SecurityMiddleware',
    'django.contrib.sessions.middleware.SessionMiddleware',
    'django.middleware.common.CommonMiddleware',
    'django.middleware.csrf.CsrfViewMiddleware',
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    'django.contrib.messages.middleware.MessageMiddleware',
    'django.middleware.clickjacking.XFrameOptionsMiddleware',
    'middlewares.AuthMD'
]
注册中间件

AuthMD中间件注册后,所有的请求都要走AuthMD的process_request方法。

如果URL在黑名单中,则返回This is an illegal URL的字符串;

访问的URL在白名单内或者session中有user用户名,则不做阻拦走正常流程;

正常的URL但是需要登录后访问,让浏览器跳转到登录页面。

注:AuthMD中间件中需要session,所以AuthMD注册的位置要在session中间的下方。 

 

附:Django请求流程图

原文地址:https://www.cnblogs.com/clbao/p/9791453.html