浅谈装饰器

 1 def foo(func):
    def inner(*args, **kwargs):
        print("主函数运行之前")
        r = func(*args, **kwargs)
        # return r, "主函数返回值"

    return inner


def test(*args, **kwargs):
    return max(args)


test = foo(test)

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):的返回值呢？看下面的代码：

 

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, ))

 

此段代码的结果为：

 

主函数运行之前
(4, '主函数返回值')

 

此时用变量r指向主函数的返回值，在return r 利用print(test(2, 3, 4, ))中的print（）就打印出了主函数的返回值。其实这也就是装饰器的实现原理，既不改变原代码，调用方式也没有改变却实现了调用主函数并在装饰器

中增加了新的功能，比如验证权限等。还有一种装饰器叫做类装饰器，还请新人们多多求索，努力学习。

-----------------有写的不好的地方还请各位指点一二，本人也是一个新人