Flask - 自定义装饰器破坏路由

Question

我有以下 Flask 路由和自定义助手：

from spots import app, db
from flask import Response
import simplejson as json


def json_response(action_func):
    def create_json_response(*args, **kwargs):
        ret = action_func(*args, **kwargs)
        code = 200
        if len(ret) == 2:
            code = ret[0]
            resp = ret[1]
        else:
            resp = ret[0]
        return Response(
            response=json.dumps(resp, indent=4),
            status=code,
            content_type='application/json'
        )

    return create_json_response


@app.route('/test')
@json_response
def test():
    return 400, dict(result="Test success")


@app.route('/')
@json_response
def home():
    return 200, dict(result="Home success")

我希望 /test 的 GET 请求返回类似 {"result":"Test success"} 的内容，但事实并非如此。相反，任何请求似乎都匹配最后一条路由，即home。为什么？

我想知道这是否是由于对 json_response 的不同调用之间缺乏绝缘造成的？

提前致谢。

Answer 1

正如Видул Петров 所说，解决方案是使用functools.wraps：

import functools
def json_response(action_func):
    @functools.wraps(action_func)
    def create_json_response(*args, **kwargs):
        ...
return create_json_response

原因是 Flask 的路由系统将 URL 映射到“端点”，然后将端点映射到视图函数。端点默认为视图函数的__name__ 属性。在这种情况下，装饰函数被传递给app.route，因此两个规则的端点都是create_json_response，并且在两种情况下都使用了为该端点定义的最后一个视图。

functools.wraps 从原始函数中获取__name__（和其他属性）并修复此问题。在装饰包装中使用它总是一个好主意。