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 = 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)