学会使用Python中的装饰器函数

装饰器函数是Python中一种强大的特性，可以用来修改或增强函数的行为。它实际上是一个高阶函数，接受一个函数作为参数，并返回一个新的函数。

装饰器函数的语法很简单，使用@符号将装饰器函数应用到目标函数上。下面是一个简单的装饰器函数的示例：

def logger(func):
    def wrapper(*args, **kwargs):
        print(f"Calling function {func.__name__}")
        result = func(*args, **kwargs)
        print(f"Function {func.__name__} finished execution")
        return result
    return wrapper

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

在上面的例子中，logger是一个装饰器函数，接受一个函数作为参数并返回一个新函数wrapper。wrapper函数在调用目标函数之前和之后会输出一些日志信息。

通过在add函数定义之前加上@logger语法，我们使用了logger装饰器函数对add函数进行了修饰。这样，当我们调用add函数时，实际上是调用了wrapper函数，从而实现了在调用目标函数之前和之后输出日志的效果。

装饰器函数可以实现各种不同的功能。下面是几个常见的装饰器函数的例子：

1. 计时装饰器：用于计算函数执行的时间。可以通过获取函数执行前后的时间戳，并计算它们的差值来实现。

import time

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"Function {func.__name__} took {end_time - start_time} seconds to execute")
        return result
    return wrapper

@timer
def fibonacci(n):
    if n <= 1:
        return n
    else:
        return fibonacci(n-1) + fibonacci(n-2)

2. 缓存装饰器：用于缓存函数的计算结果，以避免重复计算。可以通过一个字典来保存已计算的结果。

def cache(func):
    cache_dict = {}

    def wrapper(*args):
        if args in cache_dict:
            return cache_dict[args]
        else:
            result = func(*args)
            cache_dict[args] = result
            return result
    return wrapper

@cache
def factorial(n):
    if n == 0:
        return 1
    else:
        return n * factorial(n-1)

3. 权限验证装饰器：用于验证用户是否有权限执行某个函数。可以根据用户的角色或权限级别来进行验证。

def auth(role):
    def decorator(func):
        def wrapper(*args, **kwargs):
            if role == "admin":
                return func(*args, **kwargs)
            else:
                raise PermissionError("You do not have permission to execute this function")
        return wrapper
    return decorator

@auth(role="admin")
def delete_file(file_path):
    # delete file

在上面的例子中，auth是一个带参数的装饰器函数，用于指定用户的角色。通过在调用目标函数之前验证用户的角色，可以控制用户是否具有执行该函数的权限。

使用装饰器函数可以极大地提高代码的可复用性和可维护性。通过将一些通用的功能封装成装饰器函数，在需要使用这些功能的地方直接应用装饰器，可以减少代码的重复编写，并使代码更加清晰和简洁。

然而，装饰器函数也有一些使用上的限制。例如，装饰器无法访问被修饰的函数的闭包变量，因为装饰器函数在目标函数之前被调用，而闭包变量是在目标函数的调用过程中才创建的。

此外，多个装饰器函数的顺序也可能会影响函数的行为。通常情况下，被装饰的函数的属性会被最后一个装饰器函数覆盖。

总之，装饰器函数是一种强大而灵活的特性，可以通过封装和修饰函数来实现各种有用的功能。了解和熟练掌握装饰器函数的使用对于编写高质量的Python代码是非常重要的。