深入揭秘Python装饰器的应用与执行顺序：一文搞懂装饰器的魔法

本文链接：https://blog.csdn.net/gyn_enyaer/article/details/142308183

在Python编程的魔法世界中，装饰器（Decorator）无疑是一个神奇的存在。

在这里插入图片描述

它就像是一种特殊的魔法，可以在不修改原始函数代码的情况下，为函数添加额外的功能。

然而，对于很多朋友甚至有经验的开发者来说，装饰器的应用顺序和执行顺序常常令人难以理解，感到困惑。

比如下面这段代码中的装饰器，有没有让你觉得有点…头晕眼花？

在这里插入图片描述

这都是什么乱七八糟的调用关系？

我一开始，也是被装饰器弄得晕头转向，不明所以。

不过，学习就是苦中作乐，我尝试通过一段简单的程序代码，一起来探讨下装饰器的背后的神秘面纱，彻底搞懂装饰器的奥秘。

一、遇见神奇的装饰器

让我们先来看看这段代码：

# 定义装饰器1
def check1(fn1):
    def inner1(*args, **kwargs):
        print('登录验证1')
        return fn1(*args, **kwargs)
    return inner1

# 定义装饰器2
def check2(fn2):
    def inner2(*args, **kwargs):
        print('登录验证2')
        return fn2(*args, **kwargs)
    return inner2

# 使用装饰器装饰函数
@check1
@check2
def comment(user, message):
    print(f'{user} 发表评论：{message}')

在这段代码中，我们定义了两个装饰器check1和check2，并使用它们来装饰函数comment。

乍一看，这段代码似乎并不复杂，但其中却隐藏着装饰器的奥秘。

二、装饰器的应用顺序：从下往上，层层包裹

首先，我们需要理解装饰器的应用顺序。

在Python中，多个装饰器的应用顺序是从下往上的。

这意味着，最靠近函数定义的装饰器会最先应用。

在我们的代码中：

@check1
@check2
def comment(user, message):
    print(f'{user} 发表评论：{message}')

等同于：

def comment(user, message):
    print(f'{user} 发表评论：{message}')

comment = check1(check2(comment))

这就像给函数穿上了一件又一件的外套，最先穿上的外套会最贴近身体。

也就是说，comment函数首先被check2装饰，然后再被check1装饰。

1. 第一次包装：应用check2

首先，check2装饰器应用于comment函数：

comment = check2(comment)

这一步中，comment函数被check2包装，返回了一个新的函数inner2。

2. 第二次包装：应用check1

接着，check1装饰器应用于check2返回的函数inner2：

comment = check1(inner2)

此时，comment函数被check1再次包装，返回了另一个新的函数inner1。

三、装饰器的执行顺序：从外到内，层层剥开

理解了装饰器的应用顺序，我们再来看装饰器的执行顺序。

当我们调用comment函数时，实际上是在调用最外层的装饰器返回的函数，即 inner1。

执行顺序是从上往下，也就是从最外层开始，逐层深入。

1. 调用comment函数

当我们执行：

comment('Alice', '这是一条评论')

实际上调用的是inner1函数。

2. 执行inner1

在inner1函数中：

def inner1(*args, **kwargs):
    print('登录验证1')
    return fn1(*args, **kwargs)

首先，打印’登录验证1’。

然后，调用fn1(*args, **kwargs)，这里的fn1实际上是inner2。

3. 执行inner2

进入inner2函数：

def inner2(*args, **kwargs):
    print('登录验证2')
    return fn2(*args, **kwargs)

首先，打印’登录验证2’。

然后，调用fn2(*args, **kwargs)，这里的fn2就是原始的comment函数。

4. 执行原始的comment函数

最后，执行原始的comment函数：

def comment(user, message):
    print(f'{user} 发表评论：{message}')

打印’Alice 发表评论：这是一条评论’。

四、完整的执行流程图解

为了更加直观地理解，我们可以将整个执行流程绘制成一个流程图：

调用 comment('Alice', '这是一条评论')
        |
        V
执行 inner1(*args, **kwargs)
        |
打印 '登录验证1'
        |
        V
执行 fn1(*args, **kwargs)  # 这里的 fn1 是 inner2
        |
        V
执行 inner2(*args, **kwargs)
        |
打印 '登录验证2'
        |
        V
执行 fn2(*args, **kwargs)  # 这里的 fn2 是原始的 comment
        |
        V
执行 comment(user, message)
        |
打印 'Alice 发表评论：这是一条评论'

五、最终的输出结果

根据上述的执行流程，我们可以得出最终的输出结果：

登录验证1
登录验证2
Alice 发表评论：这是一条评论

六、类比思考：装饰器就像洋葱和套娃

为了让大家更好地理解装饰器的应用和执行顺序，我们可以用两个生动的比喻来形容：洋葱和俄罗斯套娃。

1. 洋葱模型

想象一下，装饰器就像一层层的洋葱皮：

在这里插入图片描述

应用装饰器时，我们一层层地给函数包裹上洋葱皮，最先包裹的装饰器在最内层。

执行函数时，我们一层层地剥开洋葱皮，最外层的装饰器最先被执行。

2. 俄罗斯套娃

装饰器也可以被看作是俄罗斯套娃：

在这里插入图片描述

每个装饰器都是一个娃娃，里面可以再套一个娃娃。

最内层的娃娃是原始的函数，最外层的娃娃是最后应用的装饰器。

调用函数时，我们需要一层层地打开套娃，直到找到最内层的原始函数。

七、为什么应用顺序和执行顺序相反？

可能有人会问，为什么装饰器的应用顺序和执行顺序是相反的呢？

这是因为装饰器在应用时，函数被一层层地包装起来，形成了一个嵌套的结构。

每个装饰器都返回了一个新的函数，这些函数之间的调用关系就像是一个栈结构。

应用装饰器时，最先应用的装饰器在最内层。

执行函数时，最外层的装饰器最先被调用。

这就导致了应用顺序和执行顺序的相反性。

八、深入理解装饰器的原理

为了更深入地理解装饰器的原理，我们可以尝试自己实现一个简单的装饰器。

1. 自定义一个简单的装饰器

def simple_decorator(func):
    def wrapper(*args, **kwargs):
        print('装饰器开始作用')
        result = func(*args, **kwargs)
        print('装饰器结束作用')
        return result
    return wrapper

2. 使用装饰器

@simple_decorator
def greet(name):
    print(f'Hello, {name}!')

greet('Bob')

3. 输出结果

装饰器开始作用
Hello, Bob!
装饰器结束作用

4. 解析

装饰器simple_decorator在应用时，将函数greet替换为wrapper函数。

调用greet(‘Bob’)时，实际上是在调用wrapper(‘Bob’)。

wrapper函数先打印’装饰器开始作用’，然后调用原始的greet函数，最后再打印’装饰器结束作用’。

九、装饰器的实际应用场景

装饰器在实际开发中有着广泛的应用，例如：

权限验证：在函数执行前进行权限检查。

日志记录：记录函数的调用信息。

性能监控：统计函数的执行时间。

通过装饰器，我们可以在不修改原始函数的情况下，为函数添加额外的功能，提高代码的可复用性和可维护性。

十、结语

装饰器是Python中非常强大且灵活的特性，理解装饰器的应用顺序和执行顺序是掌握装饰器的关键。

记住哦，装饰器的应用顺序是从下往上，执行顺序是从上往下。

形象地说，装饰器就像是给函数穿衣服，先穿的衣服在里面，调用函数时，最外层的衣服最先被处理。

和我一起学习吧，在编程实践中理解内在逻辑，才会对装饰器有更加深刻的理解。