python设计模式9-装饰器模式

1. 装饰器模式(Decorator)介绍

UML类图

装饰器模式（Decorator Pattern）允许向一个现有的对象添加新的功能，同时又不改变其结构。这种类型的设计模式属于结构型模式，它是作为现有的类的一个包装。

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

• 意图：动态地给一个对象添加一些额外的职责。就增加功能来说，装饰器模式相比生成子类更为灵活。

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

• 缺点：多层装饰比较复杂。

2. 示例

• 下面的实例来演示装饰器模式的用法。其中，我们将把一个形状装饰上不同的颜色，同时又不改变形状类。
• 创建一个 Shape 接口和实现了 Shape 接口的实体类。然后我们创建一个实现了 Shape 接口的抽象装饰类 ShapeDecorator，并把 Shape 对象作为它的实例变量。RedShapeDecorator 是实现了 ShapeDecorator 的实体类。

'''Decorator Pattern with Python Code
'''

from abc import abstractmethod, ABCMeta


# 创建一个形状接口
class Shape(metaclass=ABCMeta):
    @abstractmethod 
    def draw(self):
        pass


# 创建圆形实体类
class Circle(Shape):
    def draw(self): 
        print("Shape: Circle")


# 创建矩形实体类
class Rectangle(Shape):
    def draw(self):
        print("Shape: Rectangle")


# 创建实现了 Shape 接口的抽象装饰类
class ShapeDecorator(Shape):
    decoratedShape = None
 
    def __init__(self, decoratedShape):
        self.decoratedShape = decoratedShape
    
    def draw(self):
       self.decoratedShape.draw()


# 创建扩展了 ShapeDecorator 类的实体装饰类
class RedShapeDecorator(ShapeDecorator):
    def __init__(self, decoratedShape):
       ShapeDecorator.__init__(self, decoratedShape)   
 
    def draw(self):
       self.decoratedShape.draw()   
       self.setRedBorder(self.decoratedShape)

    def setRedBorder(self, decoratedShape):
        print("Border Color: Red")


class Client(object):
    def main(self):
        circle = Circle()
        redCircle = RedShapeDecorator(Circle())
        redRectangle = RedShapeDecorator(Rectangle())

        print("Circle with normal border")
        circle.draw()

        print("\nCircle of red border")
        redCircle.draw()

        print("\nRectangle of red border")
        redRectangle.draw()
   

if __name__ == '__main__':
    Client().main()

输出：

# ./Decorator.py
Circle with normal border
Shape: Circle

Circle of red border
Shape: Circle
Border Color: Red

Rectangle of red border
Shape: Rectangle
Border Color: Red

参考：
https://www.runoob.com/design-pattern/decorator-pattern.html