如果有AOP的编程经验,理解Python的装饰器就是分分钟的事。既然是装饰器,那么对被装饰的对象来说,一定是功能得到了增强,按方法能增强的地方进行划分,又可以分为以下四类:
1. 方法调用前;
2. 方法调用后;
3. 方法调用前后(环绕);
4. 方法调用异常;
我们以一个简单的加法运算来进行说明,按方法的增强点依次进行功能增强,首先看加法的代码,非常简单:
def add (a, b):
return a + b
1. 方法调用前
在调用之前检查方法的参数是否合规,或者判断是否有相应的执行权限,这都是常用的手法,本例中检验第一个参数是否为正数,如果是,则执行加法,否则抛出错误,那我们可以定义如下的装饰器:
def before(func):
# 只能通过参数声明的方式获取参数
def check(a, *args):
# 如果小于0,抛出异常
if(a < 0):
raise Exception('a is less than zero!')
else:
return func(a, *args)
# 记住,返回的一定是函数
return check
在这里,最难的就是怎么获取方法的参数,不要期待有反射的方式,更不要企图通过方法的属性可以获取到参数(这句话我说错了,还真有这样的方法),一定要通过变长参数的方式(元组与字典)才能获取到所有的实参。
现在继续执行添加注解与调用方式,就会出现你想要的结果,如下:
&