博客

Python Django中间件，中间件函数，全局异常处理操作示例

  • Post category:Python

Python Django中间件是Django框架提供的一个强大的工具，用于在处理请求和响应之前，前后添加一些处理函数，可以用于权限验证、请求预处理、日志等功能。本攻略将详细讲解Python Django中间件、中间件函数和全局异常处理的操作方法。

Python Django中间件

Django中间件可以理解为在Django的请求-响应处理中插入的处理过程，实现了一个类似于Java中的Filter的机制。Django中间件可以分为全局中间件、局部中间件和视图中间件。

全局中间件

全局中间件是在settings.py文件中MIDDLEWARE下添加的Middleware类，对整个Django项目的所有请求和响应生效。示例如下：

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',
]

局部中间件

局部中间件指的是只针对某个app或某个view实现处理的中间件。定义一个局部中间件需要创建一个middleware.py文件，并在其中编写自己的Middleware类，然后在需要的时候在views.py文件中导入并使用该类。示例如下：

# middleware.py
class MyMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        print("中间件处理函数")
        response = self.get_response(request)
        return response


# views.py
from myapp.middleware import MyMiddleware

def my_view(request):
    middleware = MyMiddleware()
    response = middleware(request)
    return HttpResponse(response)

视图中间件

视图中间件是专门针对某个视图函数编写的中间件。定义视图中间件需要添加对应的装饰器，并在该视图函数内部进行处理。示例如下：

def my_view(request):
    print("执行视图函数")
    return HttpResponse("Hello World!")

def my_middleware(view_func):
    def wrapper(request):
        print("执行中间件函数")
        response = view_func(request)
        return response
    return wrapper

my_view = my_middleware(my_view)

Python Django中间件函数

Django中间件给我们提供了一个实现前置和后置处理的机制，中间件函数作为Django中间件中的核心组成部分，可以做很多有意义的事情，如添加头部信息、记录日志等。下面将通过演示代码来介绍如何实现一个中间件函数。

class MyMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        print("前置处理函数")
        response = self.get_response(request)
        print("后置处理函数")
        return response

上面代码中，MyMiddleware中的__init__方法用于获取视图函数的调用的响应，然后将其保存在self.get_response属性中。接着，__call__方法作为中间件的主要处理逻辑函数，其实现逻辑是调用get_response函数，即视图函数，并返回处理后的结果。在这个过程中，可以添加前置处理和后置处理逻辑。

Python Django全局异常处理

Django中的全局异常处理是在Django处理请求时，处理异常的一种方式。Django提供了一个标准的中间件，用于处理一些异常，如404(NotFound)和500(Server Error)等。在Django中，只需要将自定义的函数添加到这个标准中间件中，就可以实现对全局异常的捕获和处理。

自定义异常中间件

class MyExceptionMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        try:
            response = self.get_response(request)
            if response.status_code == 404:
                # 404状态码的处理逻辑
                pass
        except Exception as e:
            # 异常处理逻辑
            pass

在上面的代码中，我们自定义了一个中间件类MyExceptionMiddleware，并实现了__call__方法，用来捕获请求和响应。在__call__方法的try语句中调用request处理函数，并对状态码进行判断。如果状态码为404则说明页面没有找到，需要处理该情况。如果try语句中存在异常，则会跳转至except语句中进行处理。

将自定义的异常中间件添加到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',

    # 自定义的中间件
    'myproject.middleware.MyExceptionMiddleware',
]

在上面的代码中，我们将自定义的中间件类MyExceptionMiddleware添加到了MIDDLEWARE的最后一个位置，保证该中间件最后被调用。

Tags: