因为你像一个函数一样调用装饰器,它需要返回另一个函数,它是实际的装饰器:
def my_decorator(param):
def actual_decorator(func):
print("Decorating function {}, with parameter {}".format(func.__name__, param))
return function_wrapper(func) # assume we defined a wrapper somewhere
return actual_decorator
外部函数将被赋予您明确传递的任何参数,并应返回内部函数.内部函数将传递函数进行修饰,并返回修改后的函数.
通常,您希望装饰器通过将其包装在包装函数中来更改函数行为.这是一个示例,可以选择在调用函数时添加日志记录:
def log_decorator(log_enabled):
def actual_decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
if log_enabled:
print("Calling Function: " + func.__name__)
return func(*args, **kwargs)
return wrapper
return actual_decorator
functools.wraps调用将名称和文档字符串等内容复制到包装函数中,使其更类似于原始函数.
用法示例:
>>> @log_decorator(True)
... def f(x):
... return x+1
...
>>> f(4)
Calling Function: f
5