1 def foo(func): 2 def inner(*args, **kwargs): 3 print("主函数运行之前") 4 r = func(*args, **kwargs) 5 # return r, "主函数返回值" 6 7 return inner 8 9 10 def test(*args, **kwargs):11 return max(args)12 13 14 test = foo(test)15 16 print(test(2, 3, 4, ))
上面的代码等价于下面的代码:
def foo(func): def inner(*args, **kwargs): print("主函数运行之前") r = func(*args, **kwargs) # return r, "主函数返回值" return inner @foo def test(*args, **kwargs): return max(args) # test = foo(test) print(test(2, 3, 4, ))
两者的运行结果也都是一样的:
主函数运行之前
None
也许你会很奇怪,为什么用print(test(2, 3, 4, ))会返回None,我当时也是一头雾水,下面我就解释一下这是为什么,这还是跑不了装饰器原理,一定要理解其原理。
以上面的两段代码为例,@foo替代了test = foo(test)
test = foo(test)的意思就是把函数test当做参数穿入函数foo中,那么foo的形参func(形参就相当于变量)就指向了函数test(也就是指向了test函数的内存地址),然后foo函数最下面的return返回inner
这些都是在执行“test = foo(test)”等号后面的,接下来执行等号赋值,把foo(test)结果也就是inner(内函数)赋值给等号左边的test,注意,此时等号左右两端的test完全不同,分别指向两个函数,
这也就是为什么我们print(test(2, 3, 4 ))后,打印None,实际上我们是执行了print(inner(2, 3, 4)),其并无返回值。那我们怎么打印出主函数也就是def test(*args, **kwargs):的返回值呢?看下面的代码:
1 def foo(func): 2 def inner(*args, **kwargs): 3 print("主函数运行之前") 4 r = func(*args, **kwargs) 5 return r, "主函数返回值" 6 7 return inner 8 9 10 @foo11 def test(*args, **kwargs):12 return max(args)13 14 15 # test = foo(test)16 17 print(test(2, 3, 4, ))
此段代码的结果为:
主函数运行之前
(4, '主函数返回值')
此时用变量r指向主函数的返回值,在return r 利用print(test(2, 3, 4, ))中的print()就打印出了主函数的返回值。其实这也就是装饰器的实现原理,既不改变原代码,调用方式也没有改变却实现了调用主函数并在装饰器
中增加了新的功能,比如验证权限等。还有一种装饰器叫做类装饰器,还请新人们多多求索,努力学习。
原文出处:https://www.cnblogs.com/tttzqf/p/9386104.html
共同学习,写下你的评论
评论加载中...
作者其他优质文章