检查Python函数是否使用特定装饰器进行修饰

时间:2017-08-04 22:13:33

标签: python-2.7 python-decorators

我想检查Python函数是否被修饰,并将decorator参数存储在函数dict中。这是我的代码:

from functools import wraps

def applies_to(segment="all"):
    def applies(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            func.func_dict["segment"] = segment
            print func.__name__
            print func.func_dict
            return func(*args, **kwargs)
        return wrapper
    return applies

但看起来dict丢失了:

@applies_to(segment="mysegment")
def foo():
    print "Some function"


> foo() # --> Ok, I get the expected result
foo
{'segment': 'mysegment'}

> foo.__dict__ # --> Here I get empty result. Why is the dict empty?
{}

1 个答案:

答案 0 :(得分:3)

好的,感谢user2357112的线索,我找到了答案。即使有了改进

from functools import wraps

def applies_to(*segments):
    def applies(func):
        func.func_dict["segments"] = list(segments)
        @wraps(func)
        def wrapper(*args, **kwargs):
            return func(*args, **kwargs)
        return wrapper
    return applies

谢谢!