当前位置：首页 > news >正文

python的类装饰器

news 来源：原创 2025/5/10 5:36:30

装饰器不仅可以用于函数，还能作用于类。将装饰器应用于类时，其核心原理与作用于函数类似，都是通过接收一个类作为输入，然后返回一个新的类或者修改后的原类，以此来为类添加额外的功能

简单的类装饰器

def add_method(cls):
    def new_method(self):
        print("这是新添加的方法")
    cls.new_method = new_method
    return cls

@add_method
class MyClass:
    def __init__(self):
        pass

obj = MyClass()
obj.new_method()

add_method 是一个类装饰器，它接收一个类 cls 作为参数
在装饰器内部，定义了一个新的方法 new_method，并将其添加到传入的类 cls 中
最后返回修改后的类 cls
使用 @add_method 装饰 MyClass 类，相当于执行了 MyClass = add_method(MyClass)
这样 MyClass 类就拥有了新添加的 new_method 方法

带参数的类装饰器

def add_attribute(attr_name, attr_value):
    def decorator(cls):
        setattr(cls, attr_name, attr_value)
        return cls
    return decorator

@add_attribute('new_attr', 123)
class AnotherClass:
    def __init__(self):
        pass

print(AnotherClass.new_attr)

add_attribute 是一个带参数的类装饰器，它接收两个参数 attr_name 和 attr_value，用于指定要添加的属性名和属性值
add_attribute 函数返回一个装饰器函数 decorator
decorator 函数接收类 cls 作为参数，使用 setattr 函数为类 cls 添加指定的属性
最后返回修改后的类 cls

使用 @add_attribute('new_attr', 123) 装饰 AnotherClass 类，为该类添加了一个名为 new_attr、值为 123 的属性

基于类的装饰器

class CountInstances:
    def __init__(self, cls):
        self.cls = cls
        self.instances_count = 0

    def __call__(self, *args, **kwargs):
        instance = self.cls(*args, **kwargs)
        self.instances_count += 1
        print(f"已创建 {self.instances_count} 个 {self.cls.__name__} 类的实例")
        return instance

@CountInstances
class MyTestClass:
    def __init__(self):
        pass

obj1 = MyTestClass()
obj2 = MyTestClass()

CountInstances 是一个基于类的装饰器

__init__ 方法接收被装饰的类 cls 作为参数，并初始化一个实例计数器 instances_count
__call__ 方法使得 CountInstances 类的实例可以像函数一样被调用。在 __call__ 方法中，创建被装饰类的实例，同时更新实例计数器，并打印相关信息，最后返回创建的实例

使用 @CountInstances 装饰 MyTestClass 类，每次创建 MyTestClass 类的实例时，都会触发 CountInstances 类的 __call__ 方法，从而实现对实例创建次数的统计

@CountInstances 是装饰器语法，作用相当于执行了 MyTestClass = CountInstances(MyTestClass)
将 MyTestClass 类作为参数传递给 CountInstances 类的构造函数，得到一个 CountInstances 类的实例，然后将这个实例赋值给 MyTestClass

此时，MyTestClass 实际上变成了 CountInstances 类的实例，而不再是原来的 MyTestClass 类