Python 中的 @ 符号是如何工作的！-优快云博客

写在前面

Python 中的 @ 符号是一个非常强大而又灵活的功能，它代表一个叫做"装饰器"的"语法糖"。在本文中，我们将一步步地了解它的工作原理，并通过示例代码加深理解。

基本概念

在 Python 中，@ 符号通常用于函数定义之前，它被称为"装饰器"。一个最简单的例子如下:

@print  
def say_hello():  
    print("Hello， world!")  
  
say_hello()

当我们运行这段代码时，输出将是:

<function say_hello at 0x7f6a1c0c8940>

这是怎么回事?原来，@print 其实是将 say_hello 函数"装饰"或"包装"了一层 print 函数。换句话说，say_hello = print(say_hello) 被隐式地执行了。

工作原理

现在让我们更深入地探讨一下装饰器的内部机制。装饰器的工作原理可以概括为以下几个步骤:

定义装饰器函数
将装饰器应用于目标函数
在运行时替换目标函数

让我们通过一个例子来演示这个过程:

def uppercase(func):  
    def wrapper(*args， **kwargs):  
        result = func(*args， **kwargs)  
        return result.upper()  
    return wrapper  
  
@uppercase  
def say_hello(name):  
    return f"hello， {name}"  
  
print(say_hello("Alice"))  # HELLO， ALICE

定义装饰器函数 uppercase。这个函数接受一个函数 func 作为参数，并返回一个新的函数 wrapper。
使用 @uppercase 语法将 uppercase 装饰器应用于 say_hello 函数。这实际上是将 say_hello 函数传递给 uppercase 函数，并将返回值重新赋值给 say_hello。
当我们调用 say_hello("Alice") 时，实际上调用的是 wrapper 函数，而不是原始的 say_hello 函数。wrapper 函数会调用原始的 say_hello 函数，并对其返回值进行大写转换。

通过这个过程，我们成功地在不改变 say_hello 函数本身的情况下，扩展了它的功能。这就是装饰器的核心机制。

带参数的装饰器

有时，我们可能需要为装饰器添加参数。这可以通过嵌套装饰器来实现:

def repeat(n):  
    def decorator(func):  
        def wrapper(*args， **kwargs):  
            result = func(*args， **kwargs)  
            return result * n  
        return wrapper  
    return decorator  
  
@repeat(3)  
def say_hello(name):  
    return f"hello， {name}"  
  
print(say_hello("Alice"))  # hello， Alice hello， Alice hello， Alice

在这个例子中，repeat 函数是一个"装饰器工厂"，它返回一个装饰器函数 decorator。decorator 函数接受原始函数 func 作为参数，并返回一个新的 wrapper 函数。wrapper 函数在内部调用 func，并将其返回值重复 n 次。

通过 @repeat(3) 语法，我们将 say_hello 函数"装饰"到了 repeat(3) 中，从而使得 say_hello 函数的返回值被重复 3 次。

总结

在本节中，我们深入探讨了 Python 中 @ 符号的内部原理和机制。我们了解到，@ 符号只是一种语法糖，实际上会被转换为普通的函数调用和赋值操作。我们还介绍了定义装饰器函数、将装饰器应用于目标函数，以及在运行时替换目标函数的过程。通过这些知识，相信您已经对 Python 中的 @ 符号有了更深入的理解。