函数装饰器用于在源码中“标记”函数,以某种方式增强函数的行为。它是一种以另一个函数(被装饰的函数)为参数的可调用对象,可能会处理被装饰的函数并将其返回,或者将其替换为另一个函数。
装饰器严格来说只是语法糖。假如有个名为 decorate 的装饰器:
@decorate
def target():
print(‘running target()’)
上述代码效果等同于如下写法:
def target():
print(‘running target()’)
target = decorate(target)
即原来的 target 函数会被替换为 decorate(target) 返回的函数。
def deco(func):
… def inner():
… print(‘running inner()’)
… return inner
…@deco
… def target():
… print(‘running target()’)
…target()
running inner()target
<function deco..inner at 0x7ff01bad9a60>
如上述代码,deco 返回 inner 函数对象,使用 deco 装饰 target,调用被装饰的 target 实际会运行 inner。target 对象变为 inner 的引用。
装饰器有如下两大特性:
能把被装饰的函数替换成其他函数
装饰器在加载模块时立即执行
装饰器何时执行
装饰器会在被装饰的函数定义之后立即运行,这通常是在 Python 加载模块时。
参考如下 registration.py 模块:
registry = []
def register(func):
print(f’running register({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(‘running main()’)
print(‘registry ->’, registry)
f1()
f2()
f3()
if name == ‘main’:
main()
运行后输出如下:
running register(<function f1 at 0x7fbc852d43a0>)
running register(<function f2 at 0x7fbc852d4430>)
running main()
registry -> [<function f1 at 0x7fbc852d43a0>, <function f2 at 0x7fbc852d4430>]
running f1()
running f2()
running f3()
Python 加载模块后,装饰器 register 会在其他函数之前运行,将被装饰的函数(f1 和 f2)的引用添加到 registry 列表中。原本的函数 f1 和 f2,以及未被装饰的 f3,则只在 main 明确调用它们时才执行。
如果导入 registration.py 模块(不作为脚本运行),输出如下:
import registration
running register(<function f1 at 0x7f8fbd8c3b80>)
running register(<function f2 at 0x7f8fbd8c3c10>)
函数装饰器在导入模块时立即执行,而被装饰的函数只在明确调用时执行。突出了导入时和运行时之间的区别。
变量作用域规则
测试如下函数,它读取两个变量的值,一个是局部变量 a,是函数的参数;另一个是未被定义的变量 b:
def f1(a):
… print(a)
… print(b)
…f1(3)
3
Traceback (most recent call last):
File “”, line 1, in
File “”, line 3, in f1
NameError: name ‘b’ is not defined
运行时变量 a 的值正常输出,接着报出 name ‘b’ is not defined。
若先给全局变量 b 赋值,再调用 f1 函数,就不会报错:
b = 6
f1(3)
3
6
但如下代码的结果可能会让人意想不到:
b=6
def f2(a):