装饰器：代码的魔法外衣

在编程的世界里，装饰器如同一位神秘的魔法师，轻轻一挥魔杖，就能让平凡的函数焕发出耀眼的光彩。它不仅提升了代码的可读性和可维护性，更在不知不觉中，为我们的编程之旅增添了一抹亮色。

初识装饰器

装饰器，顾名思义，是用来“装饰”函数的工具。它本质上是一个函数，接受另一个函数作为参数，并返回一个新的函数。听起来有些绕，但一旦掌握，便能体会到其中的妙处。

def my_decorator(func):
    def wrapper():
        print("Something is happening before the function is called.")
        func()
        print("Something is happening after the function is called.")
    return wrapper

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

say_hello()

在上面的例子中，my_decorator就是一个装饰器。它在不修改say_hello函数内部代码的情况下，增加了额外的功能。

装饰器的魔力

装饰器的魔力在于它的“无侵入性”。它允许我们在不改变原有函数逻辑的前提下，添加新的功能。这种设计理念符合了软件开发中的“开闭原则”——对扩展开放，对修改封闭。

日志记录

在实际开发中，日志记录是一个常见的应用场景。通过装饰器，我们可以轻松地为多个函数添加日志功能。

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

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

add(3, 4)

权限验证

在Web开发中，权限验证是一个重要的环节。装饰器可以让我们简洁地实现这一功能。

def admin_required(func):
    def wrapper(*args, **kwargs):
        if not user_is_admin():
            raise Exception("You are not allowed to perform this action.")
        return func(*args, **kwargs)
    return wrapper

def user_is_admin():
    # 这里是模拟的用户验证逻辑
    return True

@admin_required
def delete_user(user_id):
    print(f"Deleting user {user_id}")

delete_user(123)

装饰器的进阶玩法

除了基本的用法，装饰器还有许多进阶的玩法，比如带参数的装饰器、类装饰器等。

带参数的装饰器

有时候，我们希望装饰器本身也能接收参数。这时，我们可以再套一层函数。

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

@repeat(num_times=4)
def greet(name):
    print(f"Hello {name}")

greet("Alice")

类装饰器

装饰器不仅可以是函数，还可以是类。类装饰器提供了更大的灵活性，可以保存状态、实现更复杂的功能。

class CountCalls:
    def __init__(self, func):
        self.func = func
        self.num_calls = 0

    def __call__(self, *args, **kwargs):
        self.num_calls += 1
        print(f"Call {self.num_calls} of {self.func.__name__}")
        return self.func(*args, **kwargs)

@CountCalls
def say_goodbye():
    print("Goodbye!")

say_goodbye()
say_goodbye()

装饰器，这代码世界的魔法外衣，以其独特的魅力，为我们的编程之旅增添了无限可能。它不仅提升了代码的优雅性和可维护性，更在潜移默化中，提升了我们的编程思维。掌握装饰器，就像掌握了一把开启新世界的钥匙，让我们在编程的道路上，走得更远、更稳。