【Python知识】Python进阶-什么是装饰器？

Python装饰器介绍

- 概述
- - 1. 基本装饰器
  - 2. 带参数的装饰器
  - 3. 带参数的函数装饰器（使用`functools.wraps`）
  - 4. 类装饰器
  - 5. 方法装饰器（在类中使用）
- 参考文献

概述

Python装饰器是一种高级功能，允许你在不修改函数或方法定义的情况下，为其添加额外的功能或行为。装饰器本质上是一个高阶函数，它接受一个函数作为参数，并返回一个新的函数（或可调用对象）。下面是一些常见的Python装饰器类型，以及每个类型的具体例子：

1. 基本装饰器

基本装饰器是最简单的装饰器类型，它可以在函数调用前后添加一些额外的逻辑。

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()

输出：

Something is happening before the function is called.
Hello!
Something is happening after the function is called.

2. 带参数的装饰器

装饰器也可以接受参数，这允许你创建更灵活的装饰器。

def repeat(num_times):def decorator_repeat(func):def wrapper(*args, **kwargs):for _ in range(num_times):func(*args, **kwargs)return wrapperreturn decorator_repeat@repeat(num_times=3)
def greet(name):print(f"Hello, {name}!")greet("Alice")

输出：

Hello, Alice!
Hello, Alice!
Hello, Alice!

3. 带参数的函数装饰器（使用`functools.wraps`）

为了保持被装饰函数的元数据（如函数名、文档字符串等），可以使用functools.wraps装饰器。

import functoolsdef debug(func):@functools.wraps(func)def wrapper(*args, **kwargs):result = func(*args, **kwargs)print(f"{func.__name__}({args!r}, {kwargs!r}) = {result!r}")return resultreturn wrapper@debug
def add(a, b):return a + bprint(add(2, 3))

输出：

add((2, 3), {}) = 5
5

4. 类装饰器

类装饰器与函数装饰器类似，但它们应用于类而不是函数。

def singleton(cls):instances = {}def get_instance(*args, **kwargs):if cls not in instances:instances[cls] = cls(*args, **kwargs)return instances[cls]return get_instance@singleton
class MyClass:def __init__(self, value):self.value = valueobj1 = MyClass(10)
obj2 = MyClass(20)print(obj1.value)  # 输出: 10
print(obj2.value)  # 输出: 10，因为obj1和obj2是同一个实例

5. 方法装饰器（在类中使用）

虽然方法本质上也是函数，但直接在类中为方法定义装饰器需要稍微不同的语法。

def method_decorator(func):def wrapper(self, *args, **kwargs):print("Method is being called")return func(self, *args, **kwargs)return wrapperclass MyClass:@method_decoratordef my_method(self):print("Inside my_method")obj = MyClass()
obj.my_method()