Python装饰器基础及运行时间

一、装饰器基础

装饰器是可调用的对象，其参数是另一个函数（被装饰的函数）。装饰器可能会处理被装饰的函数，然后把他返回，或者将其替换成另一个函数或可调用对象。

eg：decorate装饰器

@decorate
def target():
    print("Running target()")

#上面写法等同于

def target():
    print("Running target()")
target = decorate(target)

两种写法最终得出来的结果相同，但两个代码执行完毕得到的target不一定是原来那个target函数，而是被decorate(target)返回的函数。

1、装饰器通常把函数替换成另一个函数

def deco(func):
    def inner():
        print('Running inner()')
    return inner     #deco返回inner函数对象

@deco
def target():    #使用deco装饰target
    print('Running target()')

>>target()    #运行target函数,调用被装饰的target其实会运行inner
Running inner()

>>target    #审查对象，发现target现在是inner的引用
<function deco.<locals>.inner at 0x10063b598>

装饰器只是一种语法糖，装饰器可以像常规的可调用对象一样调用，其参数是另一个函数。装饰器的一大特性是，能把被装饰的函数替换为其他函数，装饰器在加载模块时会立即执行。

二、何时执行装饰器

装饰器在被装饰的函数定义后会立即执行。

eg：registration.py模块

registry = []    #registry保存被@registry装饰的函数引用

def register(func):    #register的参数是一个函数
    print("Running register(%s)" % func)     #显示被装饰的函数
    registry.append(func)    #把func存入registry
    return func    #返回func：必须返回函数；这里返回的函数与通过参数传入的一样

@register    #f1,f2被@register装饰
def f1():
    print("Running f1()")

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

def f3():    #f3没用被装饰
    print("Running f3()")

def main():    #main显示registry，然后调用f1()、f2()、f3()
    print("Running main()")
    print("Registry ->",registry)
    f1()
    f2()
    f3()

if __name__ == "__main__":
    main()    #只有把registration.py当作脚本运行时才调用main()

上述代码输出如下：

$python3 registration.py
Running register(<function f1 at 0x1000631bf8>)
Running register(<function f1 at 0x1000631c80>)
Running main()
registry -> [<function f1 at 0x1000631bf8> ,<function f1 at 0x1000631c80>]
Running f1()
Running f2()
Running f3()

加载模块后，register中有两个被装饰函数的引用：f1、f2。这两个函数，以f3，只有在main明确调用它们才执行。

扫描二维码关注公众号，回复： 5193035 查看本文章

如果导入registration.py模块（不作为脚本运行）,输出如下：

import registration
Running register(<function f1 at 0x10063b1e0>)
Running register(<function f2 at 0x10063b237>)

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

Python装饰器基础及运行时间

猜你喜欢