12、Python - 装饰器模式

基本介绍

装饰器模式（Decorator Pattern）允许向一个现有的对象添加新的功能，同时又不改变其结构。

这种模式创建了一个装饰类，用来包装原有的类，并在保持类方法签名完整性的前提下，提供了额外的功能。

常用于扩展一个类的功能，同时这个新增的功能与原本的类没有任何关联，可以动态的进行撤销。

特点：动态地给一个对象添加一些额外的职责

装饰器模式属于结构型模式。

案例图示

在蜘蛛侠被蜘蛛咬之前，他只是一个普通人，但是被蜘蛛咬了之后，他具有了一些超能力。

同时，如果超能力在未来某一天没有了，他依然还是一个普通人：

优缺点

优点：

• 装饰类和被装饰类可以独立发展，不会相互耦合，装饰模式是继承的一个替代模式，装饰模式可以动态扩展一个实现类的功能

缺点：

• 多层装饰比较复杂

代码实现

用Python实现装饰器模式：

# 超人（装饰类）
class Superman:
    def __init__(self, person) -> None:
        self.person = person

    def fly(self):
        print("%s are flying" % self.person.name)

# 普通人（被装饰类）
class Person:
    def __init__(self, name):
        self.name = name

    def eat(self):
        print("%s are eating" % self.name)
# 装饰器类
class Decorator:
    def __init__(self, person, superman) -> None:
        self.person = person
        self.superman = superman(self.person)
    
    def __getattr__(self,attr):
        try:
            return getattr(self.person, attr)
        except AttributeError:
            return getattr(self.superman, attr)
if __name__ == "__main__":
    ken = Person("ken")
    ken.eat()
    ken.fly()

    parker = Decorator(Person("Parker"), Superman)
    parker.eat()
    parker.fly()

# ken are eating
# Parker are eating
# Parker are flying