Python - 装饰器操作目标函数的属性

Question

我的目标是想出一个装饰器，它可以接受具有任意数量参数的函数并更改目标函数的属性。例如，我需要在目标函数中选择一个属性，该属性将包含以秒为单位的执行长度。 我曾尝试利用functool的wraps方法来保留属性，但没有成功。它似乎只保留了__doc__和__name__等初始属性，但我无法为不存在的属性分配新值。 我知道我可以在返回之前将持续时间属性分配给wrapper，因为func.但这仅适用于我们不必计算函数内部发生的事情的情况，就像我的情况一样。所以，这是我的代码：

import time
from functools import wraps
def my_dec(param):
def decorator(func):
@wraps(func)
def wrapper(*args, **kwargs):
start = time.time()
rs = func(*args, **kwargs)
func.duration = time.time() - start
return rs
wrapper.a = 'a'
return wrapper
return decorator

@my_dec('Dec param')
def target_func(x):
return x.upper()
target_func('test value')
print(target_func.duration)

Answer 1

分配给wrapper.duration工作...

import time
from functools import wraps
def my_dec(param):
def decorator(func):
@wraps(func)
def wrapper(*args, **kwargs):
start = time.time()
rs = func(*args, **kwargs)
wrapper.duration = time.time() - start
return rs
return wrapper
return decorator

@my_dec('Dec param')
def target_func(x):
return x.upper()
target_func('test value')
print(target_func.duration)

Answer 2

替代答案，如果你想避免在它自己的定义中引用wrapper函数：

import time
from functools import wraps
def my_dec(param):
def decorator(func):
durations = []
@wraps(func)
def wrapper(*args, **kwargs):
start = time.time()
rs = func(*args, **kwargs)
durations.append(time.time() - start)
return rs
wrapper.durations = durations
return wrapper
return decorator

@my_dec('Dec param')
def target_func(x):
return x.upper()
target_func('test value')
print(target_func.durations)