7.31 打卡 DAY 29 复习日：类的装饰器

DAY 29 复习日：类的装饰器

我们将探索一个更高级的主题：类的装饰器。我们已经知道，函数装饰器可以“装饰”函数，那么类作为更高级的封装，是否也能被“装饰”呢？答案是肯定的。

知识点回顾

1. 类的装饰器：它的作用和写法是什么。
2. 装饰器思想的进一步理解：深入领会“外部修改”和“动态性”的核心。
3. 类方法的定义：内部定义与外部定义的区别。

1. 类的装饰器：为类穿上“盔甲”

回顾一下，函数装饰器的本质是：一个接收函数作为参数，并返回一个新函数的函数。

类的装饰器逻辑与此类似，它是一个接收类作为参数，并返回一个新类的函数。通过类装饰器，我们可以在不修改类本身代码的前提下，动态地为类增加或修改功能，例如：

• 为类添加新的方法或属性。
• 修改类已有的方法（比如在 __init__ 初始化时自动打印日志）。
• 对整个类的行为进行统一管理。

类装饰器 vs. 函数装饰器：核心区别

为了更清晰地理解，让我们来看一下两者的对比：

2. 类装饰器的实现与应用

让我们通过一个实例，看看如何编写一个为类自动添加日志功能的装饰器。

# 定义一个类装饰器：class_logger
def class_logger(cls):
    """
    这个装饰器接收一个类(cls)作为参数，
    为它添加实例化日志和一个新的log方法。
    """
    # 1. 保存原始的 __init__ 方法，以便后续调用
    original_init = cls.__init__

    # 2. 定义一个新的 __init__ 方法
    def new_init(self, *args, **kwargs):
        # 在新方法中添加我们想要的功能
        print(f"[日志] 正在实例化类: {cls.__name__}")
        # 调用原始的 __init__ 方法，完成原本的初始化工作
        original_init(self, *args, **kwargs)

    # 3. 用我们新定义的方法，替换掉类原始的 __init__ 方法
    cls.__init__ = new_init

    # 4. 在外部定义一个新函数，作为将要添加到类中的新方法
    def log_message(self, message):
        print(f"[日志] {message}")

    # 5. 将外部函数“绑定”到类上，成为类的一个新方法
    cls.log = log_message
    
    # 6. 返回被修改和增强后的类
    return cls


# 使用 @ 语法糖应用类装饰器
@class_logger
class SimplePrinter:
    def __init__(self, name):
        self.name = name

    def print_text(self, text):
        print(f"{self.name}: {text}")


# --- 使用示例 ---
# 当我们实例化 SimplePrinter 时，被装饰过的 new_init 会执行
printer = SimplePrinter("Alice")

# 调用类本身的方法
printer.print_text("Hello, World!")

# 调用由装饰器动态添加的新方法
printer.log("这是一个由装饰器添加的日志方法。")

输出结果:

[日志] 正在实例化类: SimplePrinter
Alice: Hello, World!
[日志] 这是一个由装饰器添加的日志方法。

3. 动态的艺术：内部定义 vs 外部定义方法

在上面的例子中，cls.log = log_message 这行代码可能让你感到新奇。它演示了一种动态地为类添加方法的方式。

通常，我们都在 class 代码块内部用 def 来定义方法。但Python的灵活性允许我们在外部定义一个函数，然后将其“赋值”给类的一个属性，从而使其成为类的新方法。

这种“外部赋值”的动态特性，正是类装饰器能够在不侵入原代码的情况下增强其功能的关键所在。

4. 进一步理解装饰器思想：动态与外部修改

到这里，我们应该对装饰器有一个更深刻的理解：

装饰器不仅仅是代码复用的工具，它更是一种实现“动态修改”和“外部增强”的编程思想。

@decorator 这个写法其实是一个语法糖，它的本质等同于：

# @class_logger
# class SimplePrinter:
#     ...

# 上面的代码完全等价于下面这行
# 先定义 SimplePrinter 类，然后立刻把它作为参数传给装饰器函数
# 最后将装饰器返回的新类，重新赋值给 SimplePrinter
SimplePrinter = class_logger(SimplePrinter)

理解了这一点，你就明白为什么装饰器总是在函数或类定义之前写了。同时，这也意味着即使一个类已经定义好了，你仍然可以手动调用装饰器来修改它，这充分体现了其动态性。

@浙大疏锦行