且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

在烧瓶上获取查询字符串作为函数参数

更新时间:2023-02-24 09:20:36

如果您愿意编写装饰器,则一切皆有可能:

If you are willing to write a decorator, anything is possible:

from functools import wraps

def extract_args(*names, **names_and_processors):
    user_args = ([{"key": name} for name in names] +
        [{"key": key, "type": processor}
            for (key, processor) in names_and_processors.items()])

    def decorator(f):
        @wraps(f)
        def wrapper(*args, **kwargs):
            final_args, final_kwargs = args_from_request(user_args, args, kwargs)
            return f(*final_args, **final_kwargs)
        return wrapper
    return decorator if len(names) < 1 or not is_callable(names[0]) else decorator(names[0])

def args_from_request(to_extract, provided_args, provided_kwargs):
    # Ignoring provided_* here - ideally, you'd merge them
    # in whatever way makes the most sense for your application
    results = {}
    for arg in to_extract:
        result[arg["key"]] = request.args.get(**arg)
    return provided_args, results

用法:

@app.route("/somewhere")
@extract_args("gender", age=int)
def somewhere(gender, age):
    return jsonify(gender=gender, age=age)