上一篇文章介绍了 装饰器的概念。现在讲一下在程序中怎么来写装饰器。上代码:
1 def X(fun): 2 def Y(b): 3 print(b) 4 fun() 5 return Y 6 7 def test(): 8 print('OK') 9 10 test = X(test) 11 test(1)
前五行是一个闭包,因为内层函数的参数是外层函数的变量,而外层函数返回了内存函数的引用。
第10行,在调用函数X时,将函数test的引用(注意不是test(),没有小括号)作为参数传入,此时X(test)返回的是 Y函数的引用。所以第十行的结果就是test指向了Y函数的引用。所以,第11行的test()就是调用的Y函数的引用,同时Y函数中的fun()指向了第7行的test()函数。
第二种写法:
1 def X(fun): 2 def Y(b): 3 print(b) 4 fun() 5 return Y 6 7 @X #相当于 test = X(test) 8 def test(): 9 print('OK') 10 11 test(1)
@在python中叫做语法糖,而@X 与test = X(test)等价,所以上面两种方法是等价的,只是第一种方法比较直观的展示了装饰器,而第二种写法比较美观简洁。