Python中的装饰器：理解和实现装饰者模式实例

原创拼搏现实的明天。 2025-03-20 10:09 33阅读 0赞

在Python中，装饰器是一种特殊类型的函数，它允许你修改其他函数的行为。装饰器遵循"附加功能"的编程原则，因此非常适合实现设计模式中的装饰者（Decorator）模式。

装饰者模式的典型场景是：有一些功能相似但又不完全相同的类，我们可以创建一个基类（通常是抽象类），然后让这些功能相似的类继承这个基类。

下面是一个简单的Python装饰器实现，实现了添加日志功能的装饰器：

```python
# 定义基类（这里为抽象类）—Loggable
from abc import ABC, abstractmethod

class Loggable(ABC):
    @abstractmethod
    def log(self, message: str) -> None:
        pass  # 抽象方法，实际需要在子类中实现

# 实现日志装饰器——logger_decorator
def logger_decorator(func: Loggable) -> Loggable:
    def wrapper(*args, **kwargs):
        log_message = f"Calling function {func.__name__}} with args {args}, kwargs {kwargs}"
        func.log(log_message)  # 调用子类的log方法
        return func(*args, **kwargs))  # 调用原函数并返回结果

return wrapper  # 返回装饰后的函数

# 使用装饰器装饰实际的日志记录类——LoggableExample
class LoggableExample(Loggable):
    def log(self, message: str) -> None:
        print(f"{self.__class__.__name__}}: {message}")

# 装饰函数
logger_example = logger_decorator(LoggableExample)

# 测试装饰后的日志记录功能
logger_example.log("This is a test log message.")
```

在这个例子中，`Loggable`是基类，`LoggableExample`继承了这个基类。我们创建了一个装饰器`logger_decorator`，它将任何`Loggable`类型的函数进行包装，添加日志功能。

最后，我们测试了装饰后的日志记录功能，可以看到日志被正确地打印出来。