Python装饰器

装饰器

特性

1. 能把被装饰的函数替换成其他函数
2. 装饰器在加载模块时立即执行

>>> import registration
running register(<function f1 at 0x000001656ACD0E18>)
running register(<function f2 at 0x000001656AFDA378>)

#函数装饰器在导入模块时立即执行，而被装饰的函数只在明确调用时运行

registration.py

registry = []

def register(func):
    print("running register(%s)"% func)
    registry.append(func)
    return func

@register
def f1():
    print("running f1()")

@register
def f2():
    print("running f2()")

def f3():
    print("running f3()")

def main():
    print("runing mian()")
    print("registry ->", registry)
    f1()
    f2()
    f3()


if __name__ == '__main__':
    main()

#running register(<function f1 at 0x000001656ACD0E18>)
#running register(<function f2 at 0x000001656AFDA378>)
#runing mian()
#registry -> [<function f1 at 0x000001656ACD0E18>, <function f2 at 0x000001656AFDA378>]
#running f1()
#running f2()
#running f3()

Python多层装饰器执行顺序理解

@deco1
@deco2
def foo():
    print('foo')
if __name__ == '__main__':
    foo()
#修饰器本质上就是一个函数，只不过它的传入参数同样是一个函数。
#因此，依次加了deco1和deco2两个装饰器的原函数foo
#实际上相当于deco1(deco2(foo))（）。

demo.py

def deco1(func):
    print(1)
    def wrapper1():
        print(2)
        func()
        print(3)
    print(4)
    return wrapper1

def deco2(func):
    print(5)
    def wrapper2():
        print(6)
        func()
        print(7)
    print(8)
    return wrapper2

@deco1
@deco2
def foo():
    print('foo')


if __name__ == '__main__':
    foo()

结果：
5
8
1
4
2
6
foo
7
3

解析：在foo函数对象的基础上添加deco1和deco2两个装饰器，实际上等价于deco1(deco2(foo))。首先执行的是内层函数deco2(foo)，因此第一个打印值是5和8，返回wrapper2函数，wrapper2函数没有调用，即deco1(wrapper2)，故打印的值为1和4，返回wrapper1函数，由于更外层没有装饰器，因此接下来就将执行wrapper1内的内容，打印值为2，因为deco1传入的函数变量是wrapper2,此处func()调用wrapper2函数对象，打印值为6，同样的wrapper2中的func()调用foo函数对象，故打印foo,最后，回到wrapper2和wrapper1的最后一行，依次输出7和3。整个装饰器的运行过程结束。

总结一下就是，装饰的顺序按靠近函数顺序执行，从内到外装饰，调用时由外而内，执行顺序和装饰顺序相反。