如何使用Django中间件让访问更安全？🎉
嘿，朋友们！今天我们来聊聊如何在Django中使用自定义中间件保证某个文件夹的访问需要 token 验证。毕竟，有时候我们需要确保只有合法用户才能访问某些资源，对不对？😉
什么是中间件？
简单来说，中间件就是一种在 Django 应用请求和响应过程中拦截并处理请求的组件。它们在请求到达视图之前、在响应到达客户端之前能够进行各种操作。比如本次，我们将实现一个简单的 token 验证逻辑。
创建自定义中间件
首先，你需要创建一个新的 Python 文件来放置你的中间件。让我们叫它 mymiddleware.py 吧。接着，你在该文件中定义你的中间件类：
from django.http import HttpResponse
from django.utils.deprecation import MiddlewareMixin
from center.models import Usersclass UserAuthMiddleWare(MiddlewareMixin):# 访问路由需要 token 验证def process_request(self, request):print(request.path)path = request.path # 获取访问路径# 检查访问的路径是否需要 token 验证# 在这个例子里，'media' 路径需要验证if path.split('/')[1] == 'media':# 获取 header 中的 tokentoken = request.META.get('HTTP_AUTHORIZATION')try:# 查找 token 对应的用户信息userinfo = Users.objects.filter(token=token).first()except Exception as e:return HttpResponse("Please Stay Away!")# 如果用户信息不存在，拒绝访问if userinfo is None:return HttpResponse("Please Stay Away!")
在上面的代码中，我们定义了一个类 UserAuthMiddleWare ，它继承自 MiddlewareMixin。我们在 process_request 方法中实现了 token 验证逻辑。当请求路径是 media 时，我们会从请求头中提取 Authorization 字段，然后在数据库中查找这个 token 的对应用户信息。如果找不到用户或者没有提供 token，我们将返回一个无权限访问的响应。
将中间件添加到 Django 设置
接下来，我们需要让 Django 知道有这么一个中间件，打开你的 settings.py 文件，找到 MIDDLEWARE 列表，并添加你的中间件路径：
MIDDLEWARE = [...'mymiddleware.UserAuthMiddleWare',
]
Voilà！这样就完成啦！🎉 现在，当你的用户访问 ‘media’ 目录下的资源时，系统会检查他们是否已提供有效的 token。
结语
自定义中间件给了我们强大的工具来控制请求与响应的流向，记得在实际项目中，也要对你的中间件进行测试，以确保它能在各个场景下可靠地工作哦！有什么问题，欢迎在评论区讨论，我们下次见！👋
Keep coding, keep smiling! 😄
