Django自定义中间件及其实例应用

目录

• 中间件实例
  • 实现过程
• 测试中间件

Django中，中间件是在请求和响应之间执行的钩子函数。它们是Django实现某些功能的重要方式。

本文将详细介绍Django自定义中间件，包括它们的作用、如何创建和注册中间件，以及它们的应用实例。

中间件是用于在用户请求到达视图函数之前或响应到达用户之前进行预处理的。例如，我们可以使用中间件来处理一些常见的任务，比如身份验证、缓存、HTTPS重定向等。同时，中间件也可用于统计请求时间、修改请求参数、修改响应、记录日志等等。

中间件实例

本文将以一个实例讲解中间件的应用，该实例为在请求中添加HTTP头Authorization的中间件。HTTP头Authorization可用于HTTP Basic Authentication、OAuth等身份验证机制。该中间件将为用户请求添加Authorization头，头值为从Django配置中读取的密钥。

实现过程

首先，我们需要创建一个新的Django应用。在控制台中执行以下命令：

$ django-admin startapp authmiddleware

接着，在authmiddleware目录下创建一个新的middleware.py模块。middleware.py文件位于authmiddleware文件夹下，如下所示：

authmiddleware
├── __init__.py
├── middleware.py
├── migrations
└── models.py

我们将在middleware.py中编写中间件。在middleware.py文件中，我们首先需要导入Django的HttpResponse类、settings模块和Http中间件BaseMiddleware。然后，我们可以创建我们的中间件类，它必须继承自BaseMiddleware。

from django.conf import settings
from django.http import HttpResponse
from django.utils.deprecation import MiddlewareMixin

class AuthMiddleWare(MiddlewareMixin):
    def process_request(self, request):
        key = settings.AUTH_TOKEN
        header_name = 'Authorization'
        header = 'Bearer {0}'.format(key)
        request.META[header_name] = header
        return None

在上面的代码中，我们定义了一个 AuthMiddleWare 类，并使用 process_request() 方法添加了 HTTP 头 Authorization。这个方法获取密钥（AUTH_TOKEN），然后把它加入到请求的头部中。此外，我们还添加了一个前缀 "Bearer " 到密钥中，以便让客户端知道它是 Bearer Token，而不是 Basic Auth 中的密钥。

在创建好中间件后，我们需要将它添加到 Django 的中间件列表中。在 Django 的设置文件 settings.py 中，我们添加以下代码：

MIDDLEWARE = [
    ...
    'authmiddleware.middleware.AuthMiddleWare',
]

现在，我们已经成功地在中间件列表中添加了我们的中间件。

测试中间件

最后，我们进行一个简单的测试，以确保添加的 HTTP 头已经正确到达。我们可以在控制台中使用 curl 命令发出一个 GET 请求：

$ curl -I http://localhost:8000/
HTTP/1.1 200 OK
...
Authorization: Bearer YOUR_AUTH_TOKEN
...

在上面的响应中，可以看到我们的 HTTP 头 Authorization 与已设置的密钥信息。