python装饰器记录运行时间_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的引用

.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()

Running register()

Running main()

registry -> [ ,]

Running f1()

Running f2()

Running f3()

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

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

import registration

Running register()

Running register()

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