Authorization装饰器

最新推荐文章于 2024-11-03 20:27:08 发布

发呆的比目鱼

最新推荐文章于 2024-11-03 20:27:08 发布

阅读量104

点赞数

分类专栏：工具类文章标签： python

本文链接：https://blog.csdn.net/weixin_42486623/article/details/120064592

版权

工具类专栏收录该内容

13 篇文章 0 订阅

订阅专栏

Authorization装饰器

# 比如想验证一个请求的 Owllook-Api-Key值是否跟自己约定的值一样，这里写个装饰器就好
from functools import wraps
from sanic.response import json
from settings import AUTH
from utils.status import Status

def authenticator(key):
    """
    :param keys: 验证方式 Owllook-Api-Key : Maginc Key,  Authorization : Token
    :return: 返回值
    """
    def wrapper(func):
        @wraps(func)
        async def authenticate(request, *args, **kwargs):
            value = args[0].headers.get(key, None)
            if value and AUTH[key] == value:
                response = await func(request, *args, **kwargs)
                return response
            else:
                return json({'msg': Status.TOKEN_ERROR.get_msg(), 'code': Status.TOKEN_ERROR.get_code()})
        return authenticate
    return wrapper