Python设计模式——组合模式

组合模式

组合模式：

将多个对象组合成为一个树状结构，来表示“部分-整体”的逻辑上的层次结构，组合模式使得用户对当个对象和组合对象的使用具有一致性

组合模式中的角色：

• 抽象组件（Component）
• 叶子组件（Leaf）
• 复合组件（Composite）
• 客户端（Client）

组合模式的应用场景：

• 表示对象的部分-整体层次结构（特别是结构式递归的）

• 希望用户忽略组合对象与单个对象的不同，用户统一地使用组合结构中的所有对象

组合模式的优点：

• 定义了包含基本对象和组合对象的层次结构
• 简化客户端代码，即客户端可以一致地使用组合对象和单个对象
• 更容易增加新类型的组件

python代码实现：

from abc import ABCMeta, abstractmethod


class Graphic(metaclass=ABCMeta):
    @abstractmethod
    def draw(self):
        pass


class Piont(Graphic):
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __str__(self):
        return f"画了一个点：({self.x}, {self.y})"

    def draw(self):
        print(str(self))


class Line(Graphic):
    def __init__(self, p1, p2):
        self.p1 = p1
        self.p2 = p2

    def __str__(self):
        return f"画了一个线：({self.p1}, {self.p2})"

    def draw(self):
        print(str(self))


class Picture(Graphic):
    def __init__(self, graphic_list):
        self.children = []
        for elem in graphic_list:
            self.add(elem)

    def add(self, elem):
        self.children.append(elem)

    def draw(self):
        for elem in self.children:
            elem.draw()


a = Piont(1, 2)
b = Piont(2, 4)
ab = Line(a, b)
cd = Line(Piont(4, 2), (33, 51))
pic1 = Picture([a, b, ab])
pic2 = Picture([a, cd])
pic1.draw()
pic2.draw()
print("=================")
pic3 = Picture([pic1, pic2])
pic3.draw()

输出：
====================
画了一个点：(1, 2)
画了一个点：(2, 4)
画了一个线：(画了一个点：(1, 2), 画了一个点：(2, 4))
画了一个点：(1, 2)
画了一个线：(画了一个点：(4, 2), (33, 51))
=================
画了一个点：(1, 2)
画了一个点：(2, 4)
画了一个线：(画了一个点：(1, 2), 画了一个点：(2, 4))
画了一个点：(1, 2)
画了一个线：(画了一个点：(4, 2), (33, 51))