python之装饰器的两种写法

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

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