day33类的装饰器@浙大疏锦行

qq_356196952025-12-10 13:58

day33类的装饰器@浙大疏锦行

python 复制代码

# 定义类装饰器：为类添加日志功能
def class_logger(cls):
    # 保存原始的 __init__ 方法
    original_init = cls.__init__

    def new_init(self, *args, **kwargs):
        # 新增实例化日志
        print(f"[LOG] 实例化对象: {cls.__name__}")
        original_init(self, *args, **kwargs)  # 调用原始构造方法

    # 将类的 __init__ 方法替换为新方法
    cls.__init__ = new_init

    # 为类添加一个日志方法（示例）
    def log_message(self, message):
        print(f"[LOG] {message}")

    cls.log = log_message  # 将方法绑定到类，这是一种将外部函数添加为类的属性的方法
    return cls


# 定义简单打印类，应用装饰器
# 同样是语法糖的写法
@class_logger
class SimplePrinter:
    def __init__(self, name):
        self.name = name  # 构造方法：初始化名称

    def print_text(self, text):
        """简单打印方法"""
        print(f"{self.name}: {text}")


# 使用示例
printer = SimplePrinter("Alice")  # 实例化时触发装饰器的日志
printer.print_text("Hello, World!")  # 调用普通方法
printer.log("这是装饰器添加的日志方法")  # 调用装饰器新增的方法

大体上和函数的装饰器差不多，可以在类装饰器里给类添加新的方法、属性、修改init方法等。

python 复制代码

class Robot:
    def __init__(self, name):
        self.name = name

# 1. 定义一个外部函数
def say_hello(self):
    return f"你好，我是机器人 {self.name}"

# 2. 将函数赋值给类
Robot.speak = say_hello

# 3. 调用
r = Robot("Wall-E")
print(r.speak())  # 输出: 你好，我是机器人 Wall-E

同时python还支持在类外定义类方法

@浙大疏锦行