Python中如何使用装饰器（Decorator）函数

Python中的装饰器（Decorator）函数可以让我们在不修改原函数的情况下，给函数添加一些额外的功能，这种方式非常灵活和方便。在本文中，我们将探讨如何使用Python中的装饰器函数。

装饰器函数是Python中的高级特性之一，是它的语法中的一种用来扩展函数或类的功能的函数。装饰器函数可以将一个函数作为参数来接收，并返回一个新的函数，这个新的函数会替换原来的函数，从而增加新的功能。

下面是一个简单的例子，我们定义一个函数来计算两个数的和：

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

print(add(1, 2))

输出结果为：

3

现在我们想要在该函数中添加日志记录功能，我们可以使用装饰器来实现：

def logger(func):
    def wrapper(*args, **kwargs):
        print(f"Logging start for function {func.__name__}")
        result = func(*args, **kwargs)
        print(f"Logging end for function {func.__name__}")
        return result
    return wrapper

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

print(add(1, 2))

在这个例子中，我们定义了一个名为logger的装饰器函数，该函数接受一个函数作为参数，然后定义了一个内部函数wrapper来实现日志记录功能，并返回该函数。

在使用@logger装饰器时，我们将装饰器函数logger作为参数使用，这样就可以自动给add函数添加日志记录的功能。

此时我们再次运行程序，输出结果如下：

Logging start for function add
Logging end for function add
3

我们可以看到，装饰器函数logger成功地给add函数添加了日志记录的功能。

另外，我们也可以给装饰器函数传入一些参数，从而实现更加灵活的功能。例如，我们可以给logger函数添加一个参数来指定日志的等级：

def logger(level):
    def real_logger(func):
        def wrapper(*args, **kwargs):
            print(f"Logging ({level}) start for function {func.__name__}")
            result = func(*args, **kwargs)
            print(f"Logging ({level}) end for function {func.__name__}")
            return result
        return wrapper
    return real_logger

@logger(level="DEBUG")
def add(a, b):
    return a + b

print(add(1, 2))

在这个例子中，我们定义了一个带有参数level的装饰器函数logger，并返回一个真正的装饰器函数real_logger，该函数需要接收一个函数作为参数和一个等级参数level，并返回新的函数wrapper，在新函数中可以添加一些新的功能。

在使用@logger装饰器时，我们给level传入一个值，这样就可以实现不同等级的日志记录。在这个例子中，我们使用@logger(level="DEBUG")来打印debug级别的日志。

这种方式可以让装饰器函数更加灵活，可以根据不同的需求来传入不同的参数，从而实现不同的功能。

总之，装饰器函数是Python中非常强大和灵活的工具，可以为我们的代码提供更加优秀的架构和设计，让代码更加易于维护和拓展。