一般在应用装饰器时,装饰器名副其实,就是完全的装饰作用,没有任何实际功用。
但有时为了简化代码,想把一些相同的有功能的代码放在装饰器里,但是装饰器和内部函数间的切面将两者隔离开,因此需要在装饰器和内部函数上做一点小手脚,就可以达到目的:
import datetime
class ca:
def __init__(self):
self.value='initial'
self.stat='ok'
def domethod(self):
self.value='changed'
self.stat='dump'
print 'The value is %s'%self.value
print 'The stat is %s'%self.stat
return self
def backhome(self):
self.value='initial'
print 'The value is back to %s'%self.value
def setok(self):
self.stat='ok'
print 'The stat is %s'%self.stat
def dec(func):
def _dec(*a,**ka):
InOfa=ca()
targetargs=InOfa.domethod()
res=func(targetargs,*a,**ka)
res[0].setok()
return res[1]
return _dec
@dec
def thecorefunc(targetargs,date=datetime.date.today()):
print date
targetargs.backhome()
return (targetargs,date)
if __name__=='__main__':
thecorefunc()
上面代码中的被装饰函数显而易见是thecorefunc,该函数在正常状况下的写法是不含有targetargs这个参数的,它就是需要在装饰器和函数间传递的变量(对象),在这个例子中,主体函数thecorefunc需要执行类ac的backhome方法,类ac的实例在运行时一定要先执行domethod方法,且结束时要执行setok方法,但又不想把这两个方法放在函数里,而是放在装饰器中,因此就将类ac的实例InOfa作为参数传入被装饰函数内,当被装饰函数运行后,再返回该实例,且和本身的返回值作为元祖一起返回。以待装饰器继续使用。
虽然打破了切面编程的这个切面,但也千万不要称其为黑魔法哦。