python: 装饰器
装饰器基础知识:
装饰器是可调用对象,其参数为另一个函数(被装饰的函数)。装饰器可能会处理被装饰的函数,然后把它返回,或者将其替换为另一个函数或可调用对象。
装饰器的一大特性是,能把被装饰的函数替换为其他函数,另一个特性是装饰器在加载模块时立即执行。而被装饰的函数只有在明确调用时运行。
自定义clock装饰器:
import functools
import time
def clock(func):
# @functools.wraps(func)
def clocked(*args, **kwargs):
t0 = time.time()
result = func(*args, **kwargs)
t1 = time.time() - t0
name = func.__name__
arg_list = []
if args:
arg_list.append(', '.join(repr(arg) for arg in args))
if kwargs:
pairs = ['%s=%r' % (k, w) for (k, w) in sorted(kwargs.items())]
arg_list.append(', '.join(pairs))
arg_str = ', '.join(arg_list)
print('[%0.8f] %s(%s) -> %r' % (t1, name, arg_str, result))
return result
return clocked
@clock
def suma(a, b, c=1, d=2):
total = 0
return total + a + b + c + d
result = suma(4, 5, c=4, d=9)
print(suma.__name__)
输出:
[0.00000095] suma(4, 5, c=4, d=9) -> 22
clocked
该装饰器输出程序运行的时间,接受任意个定位参数,支持关键字参数,并输出程序运行的结果。
suma会作为func参数传给clock, 然后clock函数会返回clocked函数, python解释器会把clocked赋值给suma。所以现在suma保存的是clocked函数的引用。 可以看出此时输出suma函数的名字为clocked。
这是装饰器的典型行为,把装饰的函数替换为新函数,二者接受相同的参数,而且通常返回被装饰函数本该返回的值,同时还会进行额外的操作。
上述示例中遮盖了被装饰函数的__name__ 和 __doc __属性,使用
@functools.wraps(func)装饰器,把相关属性从func赋值到clocked中,此时输出:
[0.00000095] suma(4, 5, c=4, d=9) -> 22
suma
叠放装饰器:
@d1
@d2
def func():
print('f')
等同于:
def func():
print('f")
f = d1(d2(func))