复杂装饰器原理分析

在弄复杂装饰器之前我们先温习一下简单的装饰器,代码如下：

#简单装饰器温习
def zhuangshiqi(main):
    def nei(*args,**kwargs):
        print("验证!")
        return  main(*args,**kwargs)
    return nei

@zhuangshiqi
def zhu():
    print("Welcome!")
    return 12
zhu()

这段代码的意思就是从@符号开始，代码真正开始执行，

@符号后接的是装饰器的函数名,那么装饰器便执行了，

并将@下方的主函数的函数名当做一个参数传给了装饰器,之后返回了一个nei,

这时候返回的这个nei会赋给@下方的主函数zhu这个函数名,那么就相当于zhu=nei,之后执行zhu()的时候，

实际上是执行了nei(),nei的内部代码便执行了,打印了“验证！”，之后执行参数函数，并返回参数函数的返回值，代码便运行完毕了。

接下来咋们看看复杂一点的装饰器,代码如下：

#复杂装饰器
def loading():
    print("第一种登录方式！")

def loading1():
    print("第二种登录方式！")

def error():
    print("第一种错误检查方式！")

def error1():
    print("第二种错误检查方式！")


def yanzheng(han1,han2):
    def zhuangshiqi1(main):
        def nei(*args,**kwargs):
            a=han1(*args,**kwargs)
            if a != None:
                return a
            b=main(*args,**kwargs)
            if b != None:
                return b
            c=han2(*args,**kwargs)
            if c != None:
                return c
        return nei
    return zhuangshiqi1

@yanzheng(loading,error)
def jiekou1():
    print("Welcome!")

@yanzheng(loading1,error1)
def jiekou2():
    print("Welcome!")

jiekou1()
jiekou2()

这段代码就是，我先定义了4个函数，分别是两种登录方式和两种错误检查方式,加一个复杂装饰器,

接下来从@符号开始，代码真正开始执行，大家仔细看，@符号后接得是啥，是一个带两个参数的函数，

而且加括号了对不对，那么这个函数便执行了，并且把两个参数传进去了，还返回了一个zhuangshiqi1，

这时候就相当于是@zhuangshiqi1，@符号后是不是接了一个函数名，那么这个函数你可以理解成上面那个

简单的装饰器了，之后便是直接执行zhuangshiqi1，并将@下方的主函数的函数名当做一个参数传给了装饰器,之后返回了一个nei,

这时候返回的这个nei会赋给@下方的主函数的函数名,那么就相当于主函数的函数名=nei,

之后执行主函数()的时候，实际上是执行了nei(),nei的内部代码便执行了,先执行之前传进去的第一个参数对应的函数，

接着执行参数函数，最后执行之前传进去的第二个参数对应的函数,代码便执行完毕了。

(你可以将yanzheng这个函数理解成一个默认装饰器，这样你在调用这个默认装饰器时就可以为它选择先执行哪个函数最后执行哪个函数了!)

最后附上复杂装饰器运行结果: