Python 中多个装饰器执行顺序验证

Python 中多个装饰器执行顺序验证

    正在检查是否收录...

关于 Python 装饰器执行时的顺序问题,一直以来都保持粗略的理解概念:

  1. 装饰器相当于函数调用的语法糖,因此在函数执行时,会从最内层括号开始,逐层向外执行。从代码文本上看,就是距离被修饰函数越近的装饰器,越先执行
  2. 原始的装饰器会覆盖被修饰函数的__name__等元数据,需要使用functools.wraps修饰wrapper函数,保留被修饰函数的元数据
  3. Python 代码存在

    编译时

    运行时

    的区别,因此,装饰器中代码的执行顺序,不总是被装饰函数最先执行

最近遇到一些问题,对以上概念的理解有些模糊,尝试通过代码验证。执行 Python 版本为 3.9.12。

单模块执行

测试代码

#!/usr/bin/python # encoding: utf-8 def deco1(func): print('Deco 1-1') def wrapper(*args, **kwargs): print('Deco 1 - wrapper 1') r = func(*args, **kwargs) print('Deco 1 - wrapper 2') return r print('Deco 1-2') return wrapper def deco2(func): print('Deco 2-1') def wrapper(*args, **kwargs): print('Deco 2 - wrapper 1') r = func(*args, **kwargs) print('Deco 2 - wrapper 2') return r print('Deco 2-2') return wrapper def deco3(func): print('Deco 3-1') def wrapper(*args, **kwargs): print('Deco 3 - wrapper 1') r = func(*args, **kwargs) print('Deco 3 - wrapper 2') return r print('Deco 3-2') return wrapper @deco3 @deco2 @deco1 def test(): print("Test func") if __name__ == '__main__': print('Test begin') test() print('Test end') 

测试代码中,deco1、deco2 和 deco3 装饰器都是同步函数,只有输出日志是不同的,因此,可以认为输出日志的顺序,即是代码执行的顺序。

执行结果

代码执行结果如下:

Deco 1-1 Deco 1-2 Deco 2-1 Deco 2-2 Deco 3-1 Deco 3-2 Test begin Deco 3 - wrapper 1 Deco 2 - wrapper 1 Deco 1 - wrapper 1 Test func Deco 1 - wrapper 2 Deco 2 - wrapper 2 Deco 3 - wrapper 2 Test end 

执行结果分析

分析执行结果的顺序可以发现:

  1. 进入模块__main__入口之前,会先“编译”装饰器代码,因此 "Test begin" 之前会打印Deco 1-1 等日志
  2. 在“Test begin”执行后,Deco 3 - wrapper 1 -> Deco 2 - wrapper 1 -> Deco 1 - wrapper 1 依次出现,与装饰器从上到下的应用顺序相符,而“Test func”之后,Deco 1 - wrapper 2 -> Deco 2 - wrapper 2 -> Deco 3 - wrapper 2 依次执行,顺序变成后进先出
  3. 所有装饰器代码执行结束,才轮到最后的“Test end”

多个装饰器的“编译”顺序,符合“距离最近的先执行”的印象

如果调整装饰器的应用顺序,让 deco2 最先修饰 test,关键代码如下:

@deco3 @deco1 @deco2 def test(): print("Test func") 

执行时输出结果如下,可以发现“Test begin”之前的部分中,deco2最先被调用

Deco 2-1 Deco 2-2 Deco 1-1 Deco 1-2 Deco 3-1 Deco 3-2 Test begin Deco 3 - wrapper 1 Deco 1 - wrapper 1 Deco 2 - wrapper 1 Test func Deco 2 - wrapper 2 Deco 1 - wrapper 2 Deco 3 - wrapper 2 Test end 

结论 v1.0

分析上面代码的执行顺序,可以得到以下结论:

  1. 执行被装饰的函数之前,存在类似“编译”的阶段,会执行装饰器中 wrapper 函数之外的代码
  2. 多个装饰器修饰同一个函数的情况下,最外层(即写在函数代码块最上面一行)的装饰器的 wrapper 函数中,被装饰函数之前的代码最先被执行,然后是次外层的装饰器中被装饰函数之前的代码,依此类推,直到真正执行被装饰函数;如果 wrapper 函数内存在执行后的代码,则执行顺序与 wrapper 函数内被装饰函数执行前代码的执行顺序相反,可以认为全部代码的执行顺序符合deco3(deco2(deco1(test())))的函数式执行顺序,与文档说明相符

多模块执行

测试装饰器是否应用编译缓存。将装饰器代码与测试代码放在不同的模块中。

测试代码

decos.py

#!/usr/bin/python # encoding: utf-8 def deco1(func): print('Deco 1-1') def wrapper(*args, **kwargs): print('Deco 1 - wrapper 1') r = func(*args, **kwargs) print('Deco 1 - wrapper 2') return r print('Deco 1-2') return wrapper def deco2(func): print('Deco 2-1') def wrapper(*args, **kwargs): print('Deco 2 - wrapper 1') r = func(*args, **kwargs) print('Deco 2 - wrapper 2') return r print('Deco 2-2') return wrapper def deco3(func): print('Deco 3-1') def wrapper(*args, **kwargs): print('Deco 3 - wrapper 1') r = func(*args, **kwargs) print('Deco 3 - wrapper 2') return r print('Deco 3-2') return wrapper 

decorator-test.py

#!/usr/bin/python # encoding: utf-8 from decos import deco1, deco2, deco3 @deco2 def test2(): print("Test 2 func") @deco3 @deco2 @deco1 def test(): print("Test func") if __name__ == '__main__': print('Test begin') test2() # 连续执行多个被装饰函数 print("==" * 10) test() print('Test end') 

执行结果

Deco 2-1 Deco 2-2 Deco 1-1 Deco 1-2 Deco 2-1 Deco 2-2 Deco 3-1 Deco 3-2 Test begin Deco 2 - wrapper 1 Test 2 func Deco 2 - wrapper 2 ==================== Deco 3 - wrapper 1 Deco 2 - wrapper 1 Deco 1 - wrapper 1 Test func Deco 1 - wrapper 2 Deco 2 - wrapper 2 Deco 3 - wrapper 2 Test end 

结果分析

分析以上执行结果,可以发现:

  1. Deco 2-1出现了两次,可以认为多个被装饰函数一起被执行时,装饰器 wrapper 函数之外的代码会被重复执行,执行顺序符合代码中被装饰函数的声明顺序
  2. wrapper 函数内,被装饰函数前后的代码,它们的执行顺序与单模块的情况下相同

结论

Python 中的装饰器相当于语法糖,实际执行时,代码执行顺序与多层函数包裹目标函数的写法的执行顺序一致(即 deco3(deco2(deco1(test()))) )。

在简单装饰器的场景下,例如装饰器的 wrapper 函数中直接return func(**args, **kwargs)的写法,装饰器函数本身、内部 wrapper 函数中不包含额外逻辑的情况下,可以认为被装饰函数先执行并返回结果,然后越靠近被装饰函数的装饰器越先返回。

如果装饰器本身包含较复杂的逻辑,例如测试代码中 wrapper 函数执行func前后都存在其他逻辑的写法,则需要参考多层函数的执行顺序,wrapper 函数之外的代码会先于被执行代码执行,wrapper 函数中func前的代码会按从最外层装饰器到最内层的顺序执行,func后的代码则反过来,最内层装饰器的先执行,符合多层函数调用堆栈的抽象。

  • 本文作者:WAP站长网
  • 本文链接: https://wapzz.net/post-27886.html
  • 版权声明:本博客所有文章除特别声明外,均默认采用 CC BY-NC-SA 4.0 许可协议。
本站部分内容来源于网络转载,仅供学习交流使用。如涉及版权问题,请及时联系我们,我们将第一时间处理。
文章很赞!支持一下吧 还没有人为TA充电
为TA充电
还没有人为TA充电
0
0
  • 支付宝打赏
    支付宝扫一扫
  • 微信打赏
    微信扫一扫
感谢支持
文章很赞!支持一下吧
关于作者
2.8W+
9
1
2
WAP站长官方

小米服务小程序正式上线:可实时查看服务进度、费用明细

上一篇

精选 4 款基于 C# 开源、实用的工具类库,开发效率提升利器!

下一篇
评论区
内容为空

这一切,似未曾拥有

  • 复制图片
按住ctrl可打开默认菜单