大话设计模式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/
本文版权归作者和博客园共有，欢迎转载，但未经作者同意必须在文章页面明显位置给出原文连接，否则保留追究法律责任的权利。