Python 中装饰器语法糖：核心概念与实战应用_装饰器语法糖是什么-CSDN博客

本文链接：https://blog.csdn.net/qq_62774262/article/details/146041913

在 Python 编程中，装饰器（Decorator）是一种非常强大且灵活的工具，它允许我们在不修改原有函数或方法源代码的情况下，为其添加新的功能。装饰器本质上是一个函数，它接收一个函数作为参数，并返回一个新的函数，这个新函数增强了原始函数的功能。装饰器的语法糖 @ 使得装饰器的使用更加简洁和直观。本文将详细讲解 Python 中装饰器语法糖的概念、用法及相关知识，并通过丰富的示例代码帮助读者更好地理解和应用。

一、装饰器的基本概念

（一）什么是装饰器？

装饰器本质上是一个函数，它接收一个函数作为参数，并返回一个新的函数，这个新函数增强了原始函数的功能。简而言之，装饰器就是用来“装饰”原始函数的，它通过包装（wrapper）模式修改原始函数的行为。

一个简单的装饰器示例如下：

def my_decorator(func):
    def wrapper():
        print("Before the function is called.")
        res = func()
        print("After the function is called.")
        return res
    return wrapper

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

# 调用被装饰的函数
say_hello()

在这个示例中，my_decorator 是一个装饰器函数，它接收一个函数 func 作为参数，并返回一个新的函数 wrapper。wrapper 函数包裹了原有的函数，它在调用原有函数之前和之后打印了额外的信息。@my_decorator 语法糖将 say_hello 函数传递给 my_decorator，并将其返回的新函数 wrapper 赋值给 say_hello，这样调用 say_hello 函数时，实际上是调用了 wrapper 函数。

（二）装饰器的语法糖：`@`

Python 提供了装饰器的语法糖 @，使得装饰器的使用更加简洁。上面代码中的 @my_decorator 就是装饰器的语法糖，它等价于 say_hello = my_decorator(say_hello)。

二、装饰器的用法

（一）基本用法

装饰器通过 @ 符号应用在函数定义之前，例如：

@time_logger
def target_function():
    pass

等同于：

def target_function():
    pass
target_function = time_logger(target_function)

这会将 target_function 函数传递给 time_logger 装饰器，并将返回的函数重新赋值给 target_function。从而，每次调用 target_function 时，实际上是调用了经过装饰器处理后的函数。

（二）带参数的装饰器

装饰器函数也可以接受参数，例如：

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

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

greet("World")

在这个示例中，repeat 是一个接受参数的装饰器工厂。outer_wrapper 是实际的装饰器，它接收一个函数 func 作为参数。inner_wrapper 是实际调用原始函数并添加额外功能的包装函数。

（三）类装饰器

除了函数装饰器，Python 还支持类装饰器。类装饰器允许你使用类来定义装饰器逻辑。

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

    def __call__(self, *args, **kwargs):
        print(f"调用 {self.func.__name__}")
        return self.func(*args, **kwargs)

@ClassDecorator
def example():
    pass

在这个示例中，ClassDecorator 是一个类，它的 __init__ 方法接受一个函数 func。__call__ 方法使得类的实例可以像函数一样被调用。

（四）装饰器链

你可以将多个装饰器应用于同一个函数，这些装饰器会按照从外到内的顺序被调用。

@decorator_one
@decorator_two
@decorator_three
def multi_decorated():
    pass

# 等效于 decorator_one(decorator_two(decorator_three(func)))

在这个示例中，decorator_one、decorator_two 和 decorator_three 是三个装饰器。它们会按照从外到内的顺序应用于 multi_decorated 函数。

三、装饰器的实际应用

（一）日志记录

装饰器可以用于记录函数的调用信息，例如：

def log_decorator(func):
    def wrapper(*args, **kwargs):
        print(f"调用函数 {func.__name__}，参数: {args}, {kwargs}")
        result = func(*args, **kwargs)
        print(f"{func.__name__} 函数调用结束")
        return result
    return wrapper

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

result = add(3, 5)
print(result)

在这个示例中，log_decorator 装饰器用于记录 add 函数的调用信息。

（二）权限验证装饰

器可以用于权限验证，例如：

def auth_decorator(func):
    def wrapper(*args, **kwargs):
        print("验证用户权限...")
        if not is_authenticated():
            print("用户未登录，无法访问该功能。")
            return
        return func(*args, **kwargs)
    return wrapper

@auth_decorator
def admin_page():
    print("欢迎访问管理员页面！")

def is_authenticated():
    # 模拟用户登录状态
    return True

admin_page()

在这个示例中，auth_decorator 装饰器用于验证用户权限。如果用户未登录，则无法访问被装饰的函数。

四、总结

装饰器是 Python 中一个强大的工具，它允许我们在不修改原有函数或方法源代码的情况下，为其添加新的功能。装饰器的语法糖 @ 使得装饰器的使用更加简洁和直观。通过合理使用装饰器，可以提高代码的可读性和可维护性，实现更加优雅和高效的代码设计。希望本文能够帮助读者更好地理解和应用 Python 中的装饰器语法糖。

关键点回顾：