Question

以下是我的例子：

from flask import Flask

app = Flask(__name__)
def add1(f):
    def inner(*args, **kwargs):
        return str(f(*args, **kwargs))+'1'
    return inner


@app.route('/')
@add1
def hello1():
    return "hello1";

@app.route('/hello2')
@add1
def hello2():
    return "hello2";


if(__name__ =='__main__'):
    app.run()

当我运行127.0.0.1:5000时，我希望得到“hello11”，但我得到“hello21”，为什么？

Answer 1

问题是Flask按名称跟踪函数，因为传递给app.route(path)的函数都被称为inner，第二个函数（hello2）会覆盖第一个。要解决此问题，您需要将inner函数的名称更改为正在装饰的函数的名称。您可以将装饰器更改为

def add1(f):
    def inner(*args, **kwargs):
        return str(f(*args, **kwargs))+'1'
    inner.__name__ = f.__name__
    return inner

哪个可行，但不如标准库解决方案那么优雅，

from functools import wraps

def add1(f):
    @wraps(f)
    def inner(*args, **kwargs):
        return str(f(*args, **kwargs))+'1'
    return inner

wraps装饰器不仅修复了名称，还修复了文档字符串，文件和属性字典。

烧瓶中的蟒蛇装饰

1 个答案: