我想检查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?
{}
答案 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
谢谢!