Python魔法装饰器：超越编程，开启魔法之旅

2023-10-29 12:09:04

装饰器：Python编程的隐形魔法

在Python的广阔世界中，函数扮演着至关重要的角色，从简单的计算到复杂的算法，无处不在。然而，随着代码的不断扩展，维护和重用变得越来越困难。这时，一种名为装饰器的魔法工具便横空出世，它能够赋予函数新的能力，让代码焕发新的生机。

揭开装饰器的奥秘

装饰器本质上是一个函数，它可以接受另一个函数作为参数，并在执行被装饰的函数之前或之后执行一些额外的代码。这种机制为我们提供了改变函数行为的强大能力，而无需修改函数本身的代码。

实战演练：揭秘装饰器的用法

为了更好地理解装饰器的使用，我们不妨用一个简单的例子来演示。假设我们有两个函数，一个是计算两个数之和的add函数，另一个是计算两个数之积的multiply函数。如果我们想在每次调用这两个函数时都打印出函数的参数和返回值，我们可以通过以下方式使用装饰器：

def logger(func):
    def wrapper(*args, **kwargs):
        print(f"Calling {func.__name__} with args {args} and kwargs {kwargs}")
        result = func(*args, **kwargs)
        print(f"Result of {func.__name__} is {result}")
        return result
    return wrapper

@logger
def add(a, b):
    return a + b

@logger
def multiply(a, b):
    return a * b

add(1, 2)
multiply(3, 4)

输出结果如下：

Calling add with args (1, 2) and kwargs {}
Result of add is 3
Calling multiply with args (3, 4) and kwargs {}
Result of multiply is 12

在这个例子中，我们定义了一个名为logger的装饰器。它接受一个函数作为参数，并返回一个新的函数wrapper。wrapper函数在调用被装饰的函数之前和之后分别打印出函数的参数和返回值。通过使用@logger装饰器，我们无需修改add和multiply函数的代码，便实现了代码复用，提高了代码的可读性和可维护性。