我如何获得装饰师的退货？

Question

我有一个身份验证装饰器，用于验证登录令牌。 我想从装饰器返回 UID，但我不确定该怎么做？ 有什么建议吗？ （想法是我们可以使用令牌中的 UID 来访问我们服务器上的用户特定资源）

装饰器：

def check_auth(id_token):
    decoded_token = auth.verify_id_token(id_token)
    uid = decoded_token['uid']
    print(uid)
    return uid


def requires_auth(f):
    @wraps(f)
    def decorated(*args, **kwargs):
        auth = request.headers.get("Authorization")
        if not auth or not check_auth(auth):
            message = {"error": "Authorization Required"}
            resp = message
            return resp
        return f(*args, **kwargs)

    return decorated

Answer 1

装饰器的返回值将替换已装饰的 function。 如果你从装饰器返回一个 UID，你装饰的 function 将被 UID 替换。

您可以做的是将 UID 注入到 function 调用中：

def requires_auth(f):
    @wraps(f)
    def decorated(*args, **kwargs):
        auth = request.headers.get("Authorization")
        if not auth or not check_auth(auth):
            message = {"error": "Authorization Required"}
            resp = message
            return resp
        kwargs['uid'] = check_auth(auth)
        return f(*args, **kwargs)

    return decorated

这要求所有修饰函数接受一个uid参数：

@requires_auth
def decorated_function(uid):
    …