讲 Python 装饰器前,我想先举个例子,虽有点污,但跟装饰器这个话题很贴切。
谈装饰器前,还要先要明白一件事,Python 中的函数和 Java、C++不太一样,Python 中的函数可以像普通变量一样当做参数传递给另外一个函数,例如:
先来看一个简单例子,虽然实际代码可能比这复杂很多:
现在有一个新的需求,希望可以记录下函数的执行日志,于是在代码中添加日志代码:
简单装饰器
@ 语法糖
*args、**kwargs
可能有人问,如果我的业务逻辑函数 foo 需要参数怎么办?比如:
我们可以在定义 wrapper 函数的时候指定参数:
比如:
这时,你就可以把 wrapper 函数指定关键字函数:
带参数的装饰器
@use_logging(level="warn")等价于@decorator
类装饰器
functools.wraps
例子:
装饰器顺序
一个函数还可以同时定义多个装饰器,比如: