python中的装饰器decorator

1.Python编写无参数的decorator

Python的 decorator 本质上就是一个高阶函数，它接收一个函数作为参数，然后，返回一个新函数。使用 decorator 用Python提供的 @ 语法。
考察一个@log的定义：

def log(f):
def fn(x):
print('call ’ + f.name + ‘()…’)
return f(x)
return fn
对于阶乘函数，@log工作得很好：
@log
def factorial(n):
return reduce(lambda x,y: x*y, range(1, n+1))

print(factorial(10))

结果：

call factorial()…
3628800
注意：上边传的是一个参数，如果传多个参数会报错。
要让 @log 自适应任何参数定义的函数，可以利用Python的 *args 和 **kwargs，保证任意个数的参数总是能正常调用：
def log(f):
def fn(*args, **kwargs):
print('call ’ + f.name + ‘()…’)
return f(*args, **kwargs)
return fn

Python编写无参数的decorator

Python的 decorator 本质上就是一个高阶函数，它接收一个函数作为参数，然后，返回一个新函数。

使用 decorator 用Python提供的 @ 语法，这样可以避免手动编写 f = decorate(f) 这样的代码。
考察一个@log的定义：

def log(f):
def fn(x):
print('call ’ + f.name + ‘()…’)
return f(x)
return fn

对于阶乘函数，@log工作得很好：

@log
def factorial(n):
return reduce(lambda x,y: x*y, range(1, n+1))

print(factorial(10))

结果：

call factorial()…
3628800

但是，对于参数不是一个的函数，调用将报错：

@log
def add(x, y):
return x + y
print(add(1, 2))

print(add(1, 2))
Traceback (most recent call last):
File “”, line 1, in
TypeError: fn() takes 1 positional argument but 2 were given

因为 add() 函数需要传入两个参数，但是 @log 写死了只含一个参数的返回函数。

要让 @log 自适应任何参数定义的函数，可以利用Python的 args 和 *kwargs，保证任意个数的参数总是能正常调用：

def log(f):
def fn(*args, **kwargs):
print('call ’ + f.name + ‘()…’)
return f(*args, **kwargs)
return fn

任务：

请编写一个@performance，它可以打印出函数调用的时间。

2.Python编写有参数的decorator

观察到上边的 @log 装饰器：

def log(f):
def fn(x):
print('call ’ + f.name + ‘()…’)
return f(x)
return fn

发现对于被装饰的函数，log打印的语句是不能变的（除了函数名）。
如果有的函数非常重要，希望打印出’[INFO] call xxx()…’。
有的函数不太重要，希望打印出’[DEBUG] call xxx()…’。
这时，log函数本身就需要传入’INFO’或’DEBUG’这样的参数，类似这样：
@log(‘DEBUG’)
def my_func():
pass
把上面的定义翻译成高阶函数的调用，就是：

my_func = log(‘DEBUG’)(my_func)

上面的语句看上去还是比较绕，再展开一下：

log_decorator = log(‘DEBUG’)
my_func = log_decorator(my_func)
上面的语句又相当于：
log_decorator = log(‘DEBUG’)
@log_decorator
def my_func():
pass
所以，带参数的log函数首先返回一个decorator函数，再让这个decorator函数接收my_func并返回新函数，相当于是在原有的二层嵌套里面，增加了一层嵌套：
def log(prefix):
def log_decorator(f):
def wrapper(*args, **kw):
print(’[{}] {}()…’.format(prefix, f.name))
return f(*args, **kw)
return wrapper
return log_decorator

@log(‘DEBUG’)
def test():
pass
test()

执行结果：

[DEBUG] test()…

对于这种三层嵌套的decorator定义，你可以先把它拆开：

标准decorator:
def log_decorator(f):
def wrapper(*args, **kw):
print(’[{}] {}()…’.format(prefix, f.name))
return f(*args, **kw)
return wrapper
return log_decorator

返回decorator:
def log(prefix):
return log_decorator(f)