浅谈装饰器


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 @foo
11 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