装饰器的策略可以这样来概括,它是对原有函数封闭性的保持,又对原有函数进行扩展;就是说对你原有的代码不做修改,保证其功能模块不发生改变,在原有函数的基础上,再加一层,对原有函数进行扩展;
假如说有这样一件事情:def printout(num):
print "This is a Test"
上面是输出一个字符串,但是我们现在想在print之前进行一次判断,最简单的做法是:
def printout(num):
if number:
print "The number is ",num
else:
print "nothing to print"
print "This is a Test"
按照上面的方式修改,第一破坏了开发封闭原则,第二没发现有点low吗
下面介绍用装饰器来实现:
def warp(func):
def inner(num):
if num:
print "The number is ",num
else:
print "nothing to print"
return func(num)
return inner
@warp
def printout(num):
print "This is a Test"
如果传入的参数是多个,采用下面的表示方法:
def warp(func):
def inner(num, *args, **kwargs):
if num:
print "The number is ",num
else:
print "nothing to print"
return func(num, *args, **kwargs)
return inner
@warp
def printout(num, *args, **kwargs):
print "This is a Test"