python 类的装饰器

在Python中，装饰器是一个非常强大的工具，它允许我们在不修改原始函数或类代码的情况下，增加或修改它们的行为，装饰器本质上是一个接受函数或类作为参数的函数，并返回一个新的函数或类，对于类装饰器，它接受一个类作为参数，并返回一个新的类。

（图片来源网络，侵删）

下面是一个简单的类装饰器的示例：

def class_decorator(cls):
    class WrappedClass(cls):
        def new_method(self):
            print("This is a new method from decorator")
    return WrappedClass
@class_decorator
class MyClass:
    def __init__(self):
        print("This is __init__ method of MyClass")
obj = MyClass()
obj.new_method()

在这个例子中，class_decorator就是一个类装饰器，它接受一个类cls作为参数，然后定义一个新的类WrappedClass，这个新类继承了传入的类cls，并添加了一个新的方法new_method，装饰器返回这个新的类。

当我们使用@class_decorator语法糖来装饰MyClass时，实际上是将MyClass作为参数传递给class_decorator函数，然后将返回的新类赋值给MyClass，当我们创建MyClass的实例并调用new_method时，实际上是调用的WrappedClass的new_method。

这就是类装饰器的基本工作原理，类装饰器可以做的不仅仅是添加新的方法，它们可以用来修改类的属性，方法，甚至是类的继承关系，我们可以使用装饰器来为类添加单例模式，或者为类的所有方法添加日志记录功能。

需要注意的是，类装饰器不能直接修改类的__init__方法，如果需要修改__init__方法，需要在装饰器中创建一个新的__init__方法，并在新的方法中调用原始的__init__方法。

类装饰器是Python中一个非常强大的工具，它可以让我们以简洁的方式修改类的行为，由于其强大的功能，使用时需要谨慎，避免引入不必要的复杂性。