探秘Python装饰器：代码的魔法外衣

在Python的世界里，装饰器无疑是一种神奇的存在。它们如同代码的魔法外衣，在不改变原有函数结构的前提下，为其增添新的功能。今天，让我们一起揭开装饰器的神秘面纱，探索其背后的奥秘。

装饰器的起源

装饰器的概念并非凭空而来，它源于Python的“一切皆对象”哲学。在Python中，函数不仅是代码的集合，更是一个可以被传递、修改和扩展的对象。装饰器正是利用这一特性，通过高阶函数和闭包，实现对函数的装饰。

初识装饰器

装饰器的基本语法如下：

def decorator(func):
    def wrapper(*args, **kwargs):
        # 在函数执行前添加功能
        result = func(*args, **kwargs)
        # 在函数执行后添加功能
        return result
    return wrapper

@decorator
def my_function():
    pass

在这里，decorator是一个装饰器函数，wrapper是其内部定义的包装函数。通过@decorator语法，my_function被装饰，其实质是my_function = decorator(my_function)。

装饰器的妙用

日志记录

装饰器常用于日志记录，以下是一个简单的例子：

def log_decorator(func):
    def wrapper(*args, **kwargs):
        print(f"Calling {func.__name__}")
        result = func(*args, **kwargs)
        print(f"{func.__name__} returned {result}")
        return result
    return wrapper

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

add(2, 3)

运行上述代码，你将看到函数调用和返回值的日志。

权限验证

在Web开发中，装饰器可用于权限验证：

def admin_required(func):
    def wrapper(*args, **kwargs):
        if not user.is_admin:
            raise Exception("Admin access required")
        return func(*args, **kwargs)
    return wrapper

@admin_required
def delete_user(user_id):
    pass

这样，只有管理员才能调用delete_user函数。

缓存机制

装饰器还可以实现简单的缓存机制：

def cache_decorator(func):
    cache = {}
    def wrapper(*args):
        if args in cache:
            return cache[args]
        result = func(*args)
        cache[args] = result
        return result
    return wrapper

@cache_decorator
def fibonacci(n):
    if n < 2:
        return n
    return fibonacci(n - 1) + fibonacci(n - 2)

print(fibonacci(10))

通过缓存，大大提高了计算效率。

装饰器的进阶

带参数的装饰器

有时我们需要对装饰器进行配置，这时可以使用带参数的装饰器：

def repeat(times):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for _ in range(times):
                result = func(*args, **kwargs)
            return result
        return wrapper
    return decorator

@repeat(3)
def greet(name):
    print(f"Hello, {name}")

greet("Alice")

类装饰器

装饰器不仅可以是函数，还可以是类：

class LogDecorator:
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print(f"Calling {self.func.__name__}")
        result = self.func(*args, **kwargs)
        print(f"{self.func.__name__} returned {result}")
        return result

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

multiply(2, 3)

装饰器是Python中一种强大的工具，它让我们在不修改函数内部代码的情况下，灵活地扩展函数功能。通过本文的探索，相信你对装饰器有了更深入的理解。不妨在日常编程中多加实践，让装饰器为你的代码增添更多魔法色彩。