74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
"""
|
|
实例
|
|
汤宝宝成为了上市集团老总，他们集团有好多好多家分公司...
|
|
"""
|
|
|
|
from abc import ABC, abstractmethod
|
|
|
|
|
|
# 抽象类
|
|
class Company(ABC):
|
|
name: str
|
|
children: list["Company"]
|
|
|
|
@abstractmethod
|
|
def show(self, prefix: str) -> None:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def add(self, company: "Company") -> None:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def remove(self, company: "Company") -> None:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_child(self, index: int) -> "Company":
|
|
pass
|
|
|
|
|
|
# 实现类
|
|
class ConcreteCompany(Company):
|
|
name: str
|
|
children: list[Company]
|
|
|
|
def __init__(self, name: str) -> None:
|
|
self.name = name
|
|
self.children = []
|
|
|
|
def show(self, prefix: str) -> None:
|
|
print(f"{prefix}{self.name}")
|
|
for child in self.children:
|
|
child.show(f"{prefix}-> ")
|
|
|
|
def add(self, company: Company) -> None:
|
|
self.children.append(company)
|
|
print(f"在{self.name}下开设了分公司{company.name}")
|
|
|
|
def remove(self, company: Company) -> None:
|
|
self.children.remove(company)
|
|
print(f"在{self.name}下关闭了分公司{company.name}")
|
|
|
|
def get_child(self, index: int) -> Company:
|
|
if index < 0 or index >= len(self.children):
|
|
raise IndexError("子公司索引超出范围")
|
|
return self.children[index]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("============= 组合模式 =============")
|
|
|
|
head_office = ConcreteCompany("汤氏总公司")
|
|
head_office.add(ConcreteCompany("汤氏A分公司"))
|
|
head_office.add(ConcreteCompany("汤氏B分公司"))
|
|
head_office.add(ConcreteCompany("汤氏C分公司"))
|
|
head_office.get_child(0).add(ConcreteCompany("汤氏A分公司-子公司1"))
|
|
head_office.get_child(0).add(ConcreteCompany("汤氏A分公司-子公司2"))
|
|
head_office.remove(head_office.get_child(2))
|
|
print("\n")
|
|
head_office.show("~ ")
|
|
print("\n")
|
|
head_office.get_child(0).show("~ ")
|
|
print("\n=========== 组合模式结束 ===========")
|