@print
def foo():
    pass

当我们运行以上代码后，输出结果如下：

它会打印出函数foo在内存中的位置。但是我们似乎并没有用到打印函数print()啊？为什么会这样呢？以下是原因所在：

上面的函数实际上等价于：

def foo():
    pass
hi = print(foo)

当我们运行上面的代码时，也会打印出函数foo在内存中的位置。

由于 @ 语法的存在，我们的打印功能 print(foo) 实际上被调用了 — 这就是为什么我们的 foo() 函数被打印出来的原因所在。

装饰器和被装饰对象

@something
def move():
    # 逻辑代码

这里：

something 是装饰器函数。
move 则是被装饰对象（被装饰器装饰的函数）。

action = something(move)

这里，我们的装饰器函数 something 接受被装饰对象函数 move 作为参数，因此，返回 something(move)。

一个稍微实用的示例

def move(animal):
    return animal + ' is running'

假设我们想要在不修改 move(animal) 函数源代码的前提下改变它的行为。我们可以像下面这样创建自己的装饰器函数来实现：

def add_extra(func):
    def wrapper(animal):
        return func(animal) + '!'
    return wrapper

@add_extra
def move(animal):
    return animal + ' is running'

这样，我们就可以在不修改目标函数源代码的情况下改变它的行为方式：

print(move('Dog'))
# Dog is running!

注意，在最后添加了一个感叹号 ! —— 这就是我们的装饰器函数 add_extra 的作用。

结论

希望本文能够让你加深你对装饰器的理解和应用。

—— 最后送大家一句圣人的话：道阻且长，行则将至；行而不辍，未来可期！

ITPub博客

Python魔法符号：20秒搞懂@在装饰器中的作用原理

装饰器和被装饰对象

一个稍微实用的示例

结论

最新文章