Python設計模式之組合模式原理與用法實例分析

更新時間：2019年01月11日 09:56:31 作者：Andy冉明

這篇文章主要介紹了Python設計模式之組合模式,結合具體實例形式分析了Python組合模式的相關概念、原理、定義及使用方法,需要的朋友可以參考下

本文實例講述了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
# 葉子節(jié)點
class Leaf(Component):
  def add(self,comp):
    print '不能添加下級節(jié)點'
  def remove(self,comp):
    print '不能刪除下級節(jié)點'
  def display(self, depth):
    strtemp = ''
    for i in range(depth):
      strtemp += strtemp+'-'
    print strtemp+self.name
# 枝節(jié)點
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)

運行結果如下：