Python函数——实用指南：如何使用装饰器函数来增强函数功能

在Python中，装饰器是一种特殊的函数，它可以用来增强其他函数的功能。装饰器函数可以在不修改被装饰函数源代码的情况下，为其添加额外的功能。装饰器的使用在Python的开发中非常常见，它可以提高代码的复用性和可维护性。本文将介绍如何使用装饰器函数来增强函数功能。

首先，我们先定义一个简单的函数，例如打印一句话：

def say_hello():
    print("Hello!")

现在，我们想要在调用say_hello函数之前打印一条额外的信息。我们可以使用装饰器函数来实现这个功能。装饰器函数接受一个函数作为参数，并返回一个新的函数。在新的函数中，我们可以添加额外的功能，并在最后调用原始函数。下面是一个实现了上述功能的装饰器函数的例子：

def print_before(func):
    def wrapper():
        print("Before")
        func()
    return wrapper

@print_before
def say_hello():
    print("Hello!")

在上面的例子中，print_before是一个装饰器函数，它接受一个函数作为参数并返回一个新的函数wrapper。在wrapper函数中，我们先打印一条额外的信息"Before"，然后调用原始函数func。为了应用装饰器函数，我们使用@print_before在say_hello函数之前进行装饰。

现在，我们调用say_hello函数，可以看到在打印"Hello!"之前，会先打印"Before"：

say_hello()
# Output:
# Before
# Hello!

除了添加额外的功能之外，装饰器函数还可以做其他的事情。例如，我们可以在调用原始函数之后打印一条额外的信息。下面是一个实现了这个功能的装饰器函数的例子：

def print_after(func):
    def wrapper():
        func()
        print("After")
    return wrapper

@print_before
@print_after
def say_hello():
    print("Hello!")

在上述示例中，我们定义了另一个装饰器函数print_after。在wrapper函数中，我们先调用原始函数func，然后打印"After"。最后，我们使用@print_after装饰了say_hello函数。

现在，我们调用say_hello函数，可以看到在打印"Hello!"之后，会打印"After"：

say_hello()
# Output:
# Before
# Hello!
# After

除了添加额外的功能之外，装饰器函数还可以接受参数。例如，我们可以定义一个装饰器函数来检查函数的输入参数是否是正数。下面是一个实现了这个功能的装饰器函数的例子：

def check_positive(func):
    def wrapper(x):
        if x > 0:
            func(x)
        else:
            print("Input should be a positive number")
    return wrapper

@check_positive
def print_square(x):
    print(x ** 2)

在上面的例子中，check_positive是一个装饰器函数，它接受一个函数作为参数并返回一个新的函数wrapper。在wrapper函数中，我们首先检查输入参数x是否是正数。如果是，我们调用原始函数func，否则打印一条错误信息。最后，我们使用@check_positive装饰了print_square函数。

现在，我们调用print_square函数，可以看到在输入参数是正数时，会打印平方值；在输入参数是负数时，会打印错误信息：

print_square(3)
# Output:
# 9

print_square(-3)
# Output:
# Input should be a positive number

装饰器函数在Python的开发中非常有用。它们可以用来添加额外的功能，例如日志记录，性能测试等。同时，装饰器函数可以提高代码的复用性和可维护性，使代码更加清晰和易于理解。掌握装饰器函数的使用将使你成为一个更加高效和优雅的Python开发者。