python之装饰器的两种写法

时间:2023-03-10 03:03:27
python之装饰器的两种写法

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

 def X(fun):
def Y(b):
print(b)
fun()
return Y def test():
print('OK') test = X(test)
test(1)

前五行是一个闭包,因为内层函数的参数是外层函数的变量,而外层函数返回了内存函数的引用。

第10行,在调用函数X时,将函数test的引用(注意不是test(),没有小括号)作为参数传入,此时X(test)返回的是 Y函数的引用。所以第十行的结果就是test指向了Y函数的引用。所以,第11行的test()就是调用的Y函数的引用,同时Y函数中的fun()指向了第7行的test()函数。

第二种写法:

 def X(fun):
def Y(b):
print(b)
fun()
return Y @X #相当于 test = X(test)
def test():
print('OK') test(1)

@在python中叫做语法糖,而@X 与test = X(test)等价,所以上面两种方法是等价的,只是第一种方法比较直观的展示了装饰器,而第二种写法比较美观简洁。