类装饰器
类有修饰器,他的逻辑:接收一个类,返回一个修改后的类。例如
1. 添加新的方法或属性(如示例中的 log 方法)。
2. 修改原有方法(如替换 __init__ 方法,添加日志)。
3. 甚至可以返回一个全新的类(继承或组合原类)。
通过类装饰器,可以在不修改类内部代码的情况下,为多个类统一添加功能(如日志、统计)
# 定义类装饰器:为类添加日志功能
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("这是装饰器添加的日志方法") # 调用装饰器新增的方法
注意点:
1.cls.log = log_message 把 log_message 函数赋值给 cls 类的 log 属性。这样,被装饰的类的所有实例都能调用 log 方法。
2.类的装饰器的用法和子类的用法很相似,子类也可以对原来的类进行修改,添加新的属性等,但是这两者有很大区别:
1. 实现机制不同:
类装饰器 :本质是一个函数,接收一个类作为参数,返回一个新的类或者修改后的原类。它是在类定义完成后对类进行处理,属于运行时操作。
子类 :通过继承机制创建一个新类,新类继承父类的属性和方法,并且可以在此基础上进行扩展或重写。它是在类定义阶段完成的,属于静态操作。
2. 代码复用和扩展性不同:
类装饰器 :更适合对多个不相关的类添加相同的功能,能提高代码的复用性。但对于复杂的层次化扩展,使用类装饰器会让代码逻辑变得复杂。
子类 :适合构建具有层次结构的类体系,通过继承关系可以清晰地表达类之间的关系,便于代码的维护和扩展。
3. 灵活性不同:
类装饰器 :可以在不修改原有类定义的情况下,动态地为类添加功能。一个类可以被多个装饰器装饰,实现功能的组合。如:
@decorator1
@decorator2
class MyClass:
pass
子类 :子类的功能在定义时就已经确定,若要修改功能,通常需要修改子类的定义。
4. 可维护性不同:
类装饰器 :当装饰器逻辑复杂时,可能会增加代码的理解难度,尤其是多个装饰器嵌套使用时。
子类 :继承关系清晰,代码的层次结构明确,便于理解和维护。
@浙大疏锦行