在软件开发中，我们经常会遇到这样的问题：如何在不修改原有代码的基础上，增加新的功能？这时，我们就需要使用一种叫做组合模式的设计模式，组合模式是一种结构型设计模式，它将对象组合成树形结构以表示"部分-整体"的层次结构，通过这种方式，我们可以在不改变原有代码的基础上，轻松地添加新的功能。
组合模式的主要角色有以下几个：
1、抽象组件(Composite):作为组合模式的根节点，它定义了一组操作，这些操作可以被其子节点执行，抽象组件可以是其他组件的容器，也可以是自己本身。
2、具体组件(ConcreteComponent):实现了抽象组件中定义的操作接口，具体组件可以是其他组件的容器，也可以是自己本身。
3、上下文(Context):为客户端提供一个环境，使得客户端可以使用与抽象组件相同的接口操作具体组件，上下文通常是一个包含对抽象组件引用的对象。
下面我们通过一个简单的例子来说明组合模式的使用：
假设我们正在开发一个图书管理系统，系统中有一个功能是添加书籍，我们需要实现一个功能强大的添加书籍的方法，这个方法应该能够处理各种类型的书籍，如小说、散文、诗歌等，为了实现这个功能，我们可以使用组合模式。
我们定义一个抽象组件，即Book,它包含了一些基本的属性和方法，如书名、作者、出版社等，我们定义几个具体的组件，如Novel、Prose、Poem,它们分别实现了Book中的属性和方法，我们定义一个上下文类Library,它包含了一个对Book对象的引用，并提供了一个方法来添加书籍。
from abc import ABC, abstractmethod
class Book(ABC):
def __init__(self, title, author):
self.title = title
self.author = author
@abstractmethod
def get_info(self):
pass
class Novel(Book):
def __init__(self, title, author, publisher):
super().__init__(title, author)
self.publisher = publisher
def get_info(self):
return f"{self.title} by {self.author}, published by {self.publisher}"
class Prose(Book):
def __init__(self, title, author, publisher):
super().__init__(title, author)
self.publisher = publisher
def get_info(self):
return f"{self.title} by {self.author}, published by {self.publisher}"
class Poem(Book):
def __init__(self, title, author, publisher):
super().__init__(title, author)
self.publisher = publisher
def get_info(self):
return f"{self.title} by {self.author}, published by {self.publisher}"
class Library:
def __init__(self):
self.books = []
def add_book(self, book):
if isinstance(book, Novel) or isinstance(book, Poem):
print(f"Adding novel: {book.get_info()}")
elif isinstance(book, Prose):
print(f"Adding prose: {book.get_info()}")
else:
print("Invalid book type")
self.books.append(book)通过这种方式，我们可以在不修改原有代码的基础上，轻松地添加新的书籍类型，如果我们需要添加更多的书籍类型，只需要创建一个新的具体组件即可，这就是组合模式的优势所在。
