python之装饰器的两种写法

可紊 提交于 2019-12-02 03:19:45

上一篇文章介绍了 装饰器的概念。现在讲一下在程序中怎么来写装饰器。上代码:

 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)等价,所以上面两种方法是等价的,只是第一种方法比较直观的展示了装饰器,而第二种写法比较美观简洁。

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!