Python设计模式之组合模式原理与用法实例分析

 #!/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 = Composi<strong style="color:transparent">来源gaodaima#com搞(代@码网</strong>te("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)