Django中间件middleware

中间件

    - 一个轻量级底层的插件系统，可以介入Django的请求和响应过程，修改Django的输入或输出

    - 每一个中间件组件是一个独立的python类。可以定义下面方法中的一个或多个

        __init__: 无需任何参数，服务器响应第一个请求的时候调用一次，用于确定是否启用当前中间件

        process_request(request): 执行视图之前被调用，在每个请求上调用，返回None或HttpResponse对象

        process_view(request, view_func, view_args, view_kwargs): 调用视图之前被调用，在每个请求上调用，返回None或HttpResponse

        process_template_response(request.reponse): 在视图刚好执行完毕之后被调用，在每个请求上调用，返回实现了render方法的响应对象

        process_response(request, response): 所有响应返回浏览器之前被调用，在每个请求上调用，返回HttpResponse对象

        process_exception(request, response, exception): 当视图抛出异常时调用，在每个请求上调用，返回一个HttpResponse对象

    - 如果你想干预哪个环节 只需编写对应的类做处理并注册到中间件调用的配置文件中

 

# RBAC中间件案例

import re
from django.shortcuts import redirect, HttpResponse
from django.conf import settings


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


class LoginMiddleware(MiddlewareMixin):

    def process_request(self, request):
        # login页面放行
        if request.path_info == '/stark11/login/':
            return None
        # 已经登录了放行
        if request.session.get('user_info'):
            return None
        # 否则返回login页面
        return redirect('/stark11/login/')


class RbacMiddleware(MiddlewareMixin):

    def process_request(self, request):
        # 1. 获取当前请求的URL
        # request.path_info
        # 2. 获取Session中保存当前用户的权限
        # request.session.get("permission_url_list')
        current_url = request.path_info

        # 当前请求不需要执行权限验证
        for url in settings.VALID_URL:
            if re.match(url, current_url):
                return None

        permission_dict = request.session.get(settings.PERMISSION_URL_DICT_KEY)
        if not permission_dict:
            return redirect('/stark11/login/')

        flag = False
        for group_id, code_url in permission_dict.items():

            for db_url in code_url['urls']:
                regax = "^{0}$".format(db_url)
                if re.match(regax, current_url):
                    request.permission_code_list = code_url['codes']
                    flag = True
                    break
            if flag:
                break

        if not flag:
            return HttpResponse('无权访问')