上一篇文章介绍了 装饰器的概念。现在讲一下在程序中怎么来写装饰器。上代码:
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)等价,所以上面两种方法是等价的,只是第一种方法比较直观的展示了装饰器,而第二种写法比较美观简洁。