大話設計模式Python實現-組合模式

組合模式(Composite Pattern):將對象組合成成樹形結構以表示“部分-整體”的層次結構,組合模式使得用戶對單個對象和組合對象的使用具有一致性.

下面是一個組合模式的demo：

#!/usr/bin/env python
# -*- coding:utf-8 -*-

__author__ = 'Andy'

"""
大話設計模式
設計模式——組合模式
組合模式(Composite Pattern):將對象組合成成樹形結構以表示“部分-整體”的層次結構,組合模式使得用戶對單個對象和組合對象的使用具有一致性.
"""

# 抽象一個組織類
class Component(object):

    def __init__(self, name):
        self.name = name

    def add(self,comp):
        pass

    def remove(self,comp):
        pass

    def display(self, depth):
        pass

# 葉子節點
class Leaf(Component):

    def add(self,comp):
        print '不能添加下級節點'

    def remove(self,comp):
        print '不能刪除下級節點'

    def display(self, depth):
        strtemp = ''
        for i in range(depth):
            strtemp += strtemp+'-'
        print strtemp+self.name


# 枝節點
class Composite(Component):

    def __init__(self, name):
        self.name = name
        self.children = []

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

    def remove(self,comp):
        self.children.remove(comp)

    def display(self, depth):
        strtemp = ''
        for i in range(depth):
            strtemp += strtemp+'-'
        print strtemp+self.name
        for comp in self.children:
            comp.display(depth+2)

if __name__ == "__main__":
    #生成樹根
    root = Composite("root")
    #根上長出2個葉子
    root.add(Leaf('leaf A'))
    root.add(Leaf('leaf B'))

    #根上長出樹枝Composite X
    comp = Composite("Composite X")
    comp.add(Leaf('leaf XA'))
    comp.add(Leaf('leaf XB'))
    root.add(comp)

    #根上長出樹枝Composite X
    comp2 = Composite("Composite XY")
    #Composite X長出2個葉子
    comp2.add(Leaf('leaf XYA'))
    comp2.add(Leaf('leaf XYB'))
    root.add(comp2)
    # 根上又長出2個葉子,C和D,D沒張昊,掉了
    root.add(Leaf('Leaf C'))
    leaf = Leaf("Leaf D")
    root.add(leaf)
    root.remove(leaf)
    #展示組織
    root.display(1)

上面類的設計如下圖：

應用場景：

在需要體現部分與整體層次的結構時

希望用戶忽略組合對象與單個對象的不同，統一的使用組合結構中的所有對象時

 

作者：Andy
出處：http://www.cnblogs.com/onepiece-andy/
本文版權歸作者和博客園共有，歡迎轉載，但未經作者同意必須在文章頁面明顯位置給出原文連接，否則保留追究法律責任的權利。