将查询参数传递给Flask decorator

时间:2014-10-19 23:48:05

标签: python flask flask-login

我正在为我的Flask服务器设置一个令牌身份验证系统,我希望能够设置一个装饰器,看起来像这样:

@app.route('/my/data/')
@requires_token_auth
def get_my_endpoint_data():
    """Return JSON data""""
    return get_data()

然后我会像/ my / data一样点击端点?token = myawesometokenvalue

我设置了我的装饰功能,如

def requires_token_auth(f):
    @wraps(f)
    def wrapped(*args, **kwargs):
        ... do stuff ...
        return f(*args, **kwargs)
return wrapped

不幸的是,'令牌'参数在args内部不可用。问题似乎是Flask通过了req.view_args而不是req.args。

*来自烧瓶的app.py *

1344         return self.view_functions[rule.endpoint](**req.view_args)

如何从包装函数内部访问查询参数?

2 个答案:

答案 0 :(得分:2)

由于这是“烧瓶查询参数装饰器”的第一个google结果,这是我最终在方法中的路径参数之上添加查询参数的解决方案:

def query_params(f):
    """
    Decorator that will read the query parameters for the request.
    The names are the names that are mapped in the function.
    """
    parameter_names = inspect.getargspec(f).args

    @wraps(f)
    def logic(*args, **kw):
        params = dict(kw)

        for parameter_name in parameter_names:
            if parameter_name in request.args:
                params[parameter_name] = request.args.get(parameter_name)

        return f(**params)

    return logic

@app.route('/hello/<uid>', methods=['GET', 'POST'])
@query_params
def hello_world(uid, name):
    return jsonify({
        'uid': uid,
        'name': name
    })

答案 1 :(得分:1)

哦,我刚刚意识到我可以像往常一样做到这一点!

token = request.args.get('token')