Python - 无法通过使用多个装饰器设置多个属性

Question

我在 flask API 路由上使用了多个装饰器，并且我在这些装饰器（functools wraps）中设置属性，但是我只能从被调用的第一个装饰器中设置属性。 我希望能够设置和引用所有被调用的装饰器的属性。

装饰器.py

from functools import wraps

def example_func1(f):
    @wraps(f)
    def decorated(*args, **kwargs):

        setattr(decorated, 'name', 'bill')

        return f(*args, **kwargs)

    return decorated

    
    def example_func2(f):
        @wraps(f)
        def decorated(*args, **kwargs):

            setattr(decorated, 'cookie', 'chocolate')
    
            return f(*args, **kwargs)
    
        return decorated

    
    def example_func3(f):
        @wraps(f)
        def decorated(*args, **kwargs):

            setattr(decorated, 'shoes', 'nike')
    
            return f(*args, **kwargs)
    
        return decorated

Flask 路线：

@app.route("/myRoute", methods=["GET", "OPTIONS"])
@example_func1
@example_func2
@example_func3
def my_route():
    print(my_route.name) # this returns 'bill' as expected
    print(my_route.cookie) # stack trace for no attribute
    print(my_route.shoes) # didn't get this far

来自堆栈跟踪的错误消息：

AttributeError: 'function' object has no attribute 'cookie'

Answer 1

您可以将setattr移到def之后，以便在您刚刚定义（并且您将返回）的decorated function 上分配属性，如下所示：

def example_func3(f):
    @wraps(f)
    def decorated(*args, **kwargs):
        return f(*args, **kwargs)
    setattr(decorated, 'shoes', 'nike')
    return decorated

我已经测试过（在烧瓶外），这对我有用。

但是当然，这引出了一个问题：为什么我需要decorated和@wraps？ 好吧，我不确定你是否这样做，这也适用于我（同样，没有烧瓶）：

from functools import wraps

def example_func1(f):
    f.name = 'bill'
    return f

def example_func2(f):
    f.cookie = 'chocolate'
    return f

def example_func3(f):
    f.shoes = 'nike'
    return f

@example_func1
@example_func2
@example_func3
def func():
    print(f'func: name={func.name}')
    print(f'func: cookie={func.cookie}')
    print(f'func: shoes={func.shoes}')

func()

（请注意，我已经简化了setattr调用）