简单介绍Python中的装饰器及其应用场景

装饰器是Python的一种语法糖，它允许我们通过在函数定义前加上@符号，来动态地修改函数的行为。装饰器本质上是一个函数，它接受一个函数作为参数，并返回一个新的函数。

装饰器的应用场景非常广泛，可以用于日志记录、性能测试、访问控制、缓存等方面。下面将介绍几个常见的装饰器应用场景及其示例。

1. 日志记录：装饰器可以用来记录函数的调用和返回结果，方便开发人员追踪程序的执行过程和输出结果。下面是一个简单的日志记录装饰器示例：

import logging

def log_decorator(func):
    def wrapper(*args, **kwargs):
        logging.info(f"Calling function {func.__name__} with arguments {args}, {kwargs}")
        result = func(*args, **kwargs)
        logging.info(f"Function {func.__name__} returned {result}")
        return result
    return wrapper

@log_decorator
def add(x, y):
    return x + y

print(add(1, 2))  # 输出: 3

2. 计时器：装饰器可以用来统计函数的执行时间，方便性能测试和优化。下面是一个简单的计时器装饰器示例：

import time

def timer_decorator(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 run")
        return result
    return wrapper

@timer_decorator
def fib(n):
    if n <= 1:
        return n
    else:
        return fib(n-1) + fib(n-2)

print(fib(30))  # 输出: 832040

3. 访问控制：装饰器可以用来限制对某些函数的访问权限，只有满足特定条件的用户才能执行该函数。下面是一个简单的访问控制装饰器示例：

def access_control_decorator(required_role):
    def decorator(func):
        def wrapper(*args, **kwargs):
            user_role = get_user_role()  # 获取当前用户角色的函数
            if user_role == required_role:
                return func(*args, **kwargs)
            else:
                raise PermissionError("Access denied")
        return wrapper
    return decorator

@access_control_decorator("admin")
def delete_user(user_id):
    # 删除用户的逻辑
    pass

delete_user(1234)  # 如果用户是管理员，函数将会被执行；否则，将会抛出PermissionError异常

4. 缓存：装饰器可以用来缓存函数的运行结果，避免重复计算，提高效率。下面是一个简单的缓存装饰器示例：

def cache_decorator(func):
    cache = {}

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

    return wrapper

@cache_decorator
def fib(n):
    if n <= 1:
        return n
    else:
        return fib(n-1) + fib(n-2)

print(fib(30))  # 输出: 832040；第二次执行时，函数会直接返回之前的缓存结果，不会重新计算

以上是几个常见的装饰器应用场景及其示例。装饰器可以帮助我们简化代码、提高代码的复用性，并在不修改原有代码的基础上，给函数增加额外的功能。在实际开发中，我们可以根据需要自定义各种装饰器，以满足不同的需求。