django校验token自定义中间件

发布时间:2024年01月12日

首先,创建一个名为 middleware.py 的文件,定义你的中间件:

from django.http import JsonResponse

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

    def __call__(self, request):
        # 在请求处理之前执行的代码

        # 获取请求中的 Token
        token = request.META.get('HTTP_AUTHORIZATION', None)

        # 校验 Token,这里你可以使用你自己的校验逻辑
        if not self.is_valid_token(token):
            return JsonResponse({'error': 'Invalid token'}, status=401)

        # 请求继续传递给下一个中间件或视图
        response = self.get_response(request)

        # 在请求处理之后执行的代码

        return response

    def is_valid_token(self, token):
        # 这里是校验 Token 的具体逻辑,可以根据你的需求进行定制
        # 返回 True 表示 Token 有效,返回 False 表示无效
        # 你可以使用任何 Token 校验库,如 `python-jose`、`PyJWT` 等

        # 示例:假设 Token 是一个简单的字符串,长度大于等于 10 就认为有效
        return token is not None and len(token) >= 10

然后,在你的 Django 项目的 settings.py 中注册这个中间件:

MIDDLEWARE = [
    # 其他中间件...
    'path.to.TokenValidationMiddleware',
    # 其他中间件...
]

确保将 ‘path.to.TokenValidationMiddleware’ 替换为你中间件的实际路径。中间件会按照在列表中的顺序顺序执行,所以确保它在其他可能影响请求处理的中间件之前或之后执行,具体取决于你的需求。
这是一个简单的例子,实际上,你可能需要使用更复杂的 Token 校验逻辑,例如使用 JSON Web Tokens(JWT)库进行签名和验证。

文章来源:https://blog.csdn.net/Ling_Ze/article/details/135550237
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。