Django——中间件

Django------中间件

中间件可以介入 Django 的请求和响应的处理过程，修改 Django 的响应数据。中间件的设计为程序开发者提供了一种无侵入式的开发方式，增强 Django 框架的健壮性。

中间件可以在 Django 处理视图的不同阶段的干预。

Django 框架中原先内置的一些中间件

python 复制代码

MIDDLEWARE = [
    # 主要对安全性访问处理请求，把 http 请求重定向为 https 请求
    'django.middleware.security.SecurityMiddleware',
    # 会话支持
    'django.contrib.sessions.middleware.SessionMiddleware',
    # 检查是否允许浏览器访问类型 ， 检查 url 是否需要添加 /
    'django.middleware.common.CommonMiddleware',
    # 跨站点防御保护
    'django.middleware.csrf.CsrfViewMiddleware',
    # 用户认证
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    # Cookie 会话支持
    'django.contrib.messages.middleware.MessageMiddleware',
    # 点击劫持保护
    'django.middleware.clickjacking.XFrameOptionsMiddleware',
]

1、项目启动的时候
2、请求进来的时候需要经过中间件之后才能到后续的路由匹配操作
3、响应到浏览器的时候需要经过中间件

1、认证登录
2、统计访问流量
3、拦截请求：可以限制IP的请求频率或者直接封禁IP

中间件的执行顺序

在请求处理之前，中间件根绝配置文件中的配置列表从上往下依次执行；

在请求处理之后，中间件根绝配置文件中的配置列表从下往上依次执行；

1、自定义中间件

定义一个中间件工厂函数，返回一个可以被调用的中间件（闭包）。

中间件的工厂函数必须接收一个可以被调用的 get_response 对象，请求处理。

python 复制代码

def simple_middleware(get_response):
    # 这里编写的代码只在 Django 项目启动的时候被执行
    def middleware(request):
        '''
        request : 和我们写的视图函数接收的request数据一致，都是浏览器的请求数据
        '''
        # 这里编写的代码，是在请求处理之前被执行
        response = get_response(request)
        # 这里编写的代码，是在请求处理之后被执行
        return response
        
    return middleware

在应用下新建一个 middleware.py 文件

python 复制代码

def simple_middleware(get_response):
    # 这里编写的代码只在 Django 项目启动的时候被执行
    print('初始化 Django')
    def middleware(request):
        '''
        request : 和我们写的视图函数接收的request数据一致，都是浏览器的请求数据
        '''
        # 这里编写的代码，是在请求处理之前被执行
        print('request 请求被处理之前')
        response = get_response(request)
        # 这里编写的代码，是在请求处理之后被执行
        print('request 请求被处理之后')
        return response
    return middleware

定义好中间件之后，需要到配置文件中的 MIDDLEWARE 列表中添加自定义的中间件路径

python 复制代码

# 自定义中间件注册
'MiddlewareApp.middleware.simple_middleware',

2、自定义中间件类

自定义中间件类需要继承 MiddlewareMixin

process_request 方法，是 Request 预处理函数；这个方法就是在请求处理之前被调用

python 复制代码

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

class MyMiddleware(MiddlewareMixin):

    def process_request(self , request):
        # request ： 用户请求的数据
        # 可以没有返回值：None 中间件就会按照正常的流程执行， 如果配置列表后面还有中间就进入下一个中间件，
        #               否则则进入 url 请求匹配
        print('process_request 预处理函数')
        # 有返回值：HttpResponse 对象，就不执行处理其他的中间件，以及不进行 url 匹配，这里整个请求停止
        #           将这里返回的 HttpResponse 对象响应给浏览器
        return HttpResponse('我是 process_request 响应的数据')

process_response 方法；这个方法是在请求处理之后被调用

python 复制代码

    def process_response(self , request , response):
        '''
        :param request: 用户请求的数据
        :param response: 是视图返回出来的 HttpResponse 对象
        :return: 必须要有返回值 ， 必须返回一个 HttpResponse 对象
                可以返回视图的 response 对象；也可以自定义 HttpResponse 对象 , 则视图中返回值无效
        '''
        print('process_response 执行')
        # return response
        return HttpResponse('我是 process_response 响应的数据')

process_view 方法；这个方法在路由匹配之后，视图响应之前

python 复制代码

    def process_view(self , request , view_func , view_args , view_kwargs):
        '''
        :param request:
        :param view_func:  请求的 url 匹配到的视图函数
        :param view_args: 是给视图传递的参数
        :param view_kwargs: 是给视图传递的参数
        :return: 可以没有返回值；可以返回值匹配到的视图函数，
                            也可以自定义 HttpResponse 对象 , 则视图函数无效
        '''
        print('process_view 执行')
        # return view_func(request)   # ok
        return HttpResponse('我是 process_view 响应的数据')

3、中间件拦截请求

1、限制 IP 访问频率

python 复制代码

visit_ip = {}
class IpMiddleware(MiddlewareMixin):
    '''
    1、 获取访问的主机IP
    2、获取到 IP 访问的当前时间
    3、时间间隔为：40s ；超过15次
    '''
    def process_request(self , request):
        # 获取用户访问的主机 IP
        ip = request.META.get('REMOTE_ADDR')
        # 获取 IP 访问的当前时间
        visit_time = time.time()
        # 数据保存格式{IP:[访问时间,......]，IP2:[访问时间,......]......}
        # 判断 IP 是否在字典中存在 ， 如果存在则这个 IP 不是第一次访问
        # 否则 这个 IP 是第一次访问
        if ip not in visit_ip:
            visit_ip[ip] = [visit_time]

        # ip 在字典中，将访问时间间隔为 大于40s的进行删除时间记录
        old_time = visit_ip.get(ip)
        for t in old_time:
            # 如果列表时间和当前访问时间超过 40s 则清除掉
            if visit_time - t > 40:
                old_time.remove(t)
        # 判断时间列表中记录是否有超过 15 次
        # 没有超过，进行记录当前的访问时间
        # 超过，限制当前 IP 访问
        if len(old_time) < 15:
            old_time.insert(0 , visit_time)
        else:
            return HttpResponse("您别累着了 ， 休息一下吧！！！！")

2、禁止 IP 访问

python 复制代码

black_ip = {}   # {IP:count}
visit_ip = {}
class IpMiddleware(MiddlewareMixin):
    '''
    1、 获取访问的主机IP
    2、获取到 IP 访问的当前时间
    3、时间间隔为：40s ；超过 5次
    4、保存用户的不良行为记录次数 ， 超过 3
    '''
    def process_request(self , request):
        # 获取用户访问的主机 IP
        ip = request.META.get('REMOTE_ADDR')
        ## 从数据库中取出ip进行判断（ip） ， 数据库中有则表示已经拉黑 ， 没有则表示正常
        # 检查 IP 是否被拉黑
        black_num = black_ip.get(ip , 0)
        if black_num > 3:
            ## 将 ip 拉黑保存到数据库中
            return HttpResponse(status=404)
        # 获取 IP 访问的当前时间
        visit_time = time.time()
        # 数据保存格式{IP:[访问时间,......]，IP2:[访问时间,......]......}
        # 判断 IP 是否在字典中存在 ， 如果存在则这个 IP 不是第一次访问
        # 否则 这个 IP 是第一次访问
        if ip not in visit_ip:
            visit_ip[ip] = [visit_time]

        # ip 在字典中，将访问时间间隔为 大于40s的进行删除时间记录
        old_time = visit_ip.get(ip)
        for t in old_time:
            # 如果列表时间和当前访问时间超过 40s 则清除掉
            if visit_time - t > 40:
                old_time.remove(t)
        # 判断时间列表中记录是否有超过 15 次
        # 没有超过，进行记录当前的访问时间
        # 超过，限制当前 IP 访问
        if len(old_time) < 5:
            old_time.insert(0 , visit_time)
        else:
            # 保存累计不良行为记录次数
            black_ip[ip] = black_num + 1
            return HttpResponse("您别累着了 ， 休息一下吧！！！！")