Python中使用组合模式来处理复杂的层次结构

from abc import ABC, abstractmethod

class CompanyComponent(ABC):
    """
    抽象基类，定义组合对象和单个对象的公共接口
    """
    @abstractmethod
    def get_name(self):
        pass

    @abstractmethod
    def get_description(self):
        pass

    @abstractmethod
    def get_children(self):
        pass

class Department(CompanyComponent):
    """
    具体的组合对象类，代表一个部门
    """
    def __init__(self, name, description):
        self._name = name
        self._description = description
        self._children = []

    def add_child(self, child):
        self._children.append(child)

    def remove_child(self, child):
        self._children.remove(child)

    def get_name(self):
        return self._name

    def get_description(self):
        return self._description

    def get_children(self):
        return self._children

class Employee(CompanyComponent):
    """
    具体的单个对象类，代表一个雇员
    """
    def __init__(self, name, description):
        self._name = name
        self._description = description

    def get_name(self):
        return self._name

    def get_description(self):
        return self._description

    def get_children(self):
        return []

# 创建公司层次结构
company = Department("pidancode.com", "一个神奇的编程公司")
department1 = Department("技术部", "负责公司的技术开发")
department2 = Department("人力资源部", "负责公司的招聘和管理")
employee1 = Employee("小明", "技术部的工程师")
employee2 = Employee("小红", "人力资源部的招聘专员")

department1.add_child(employee1)
department2.add_child(employee2)
company.add_child(department1)
company.add_child(department2)

# 遍历公司层次结构
def traverse(component, level=0):
    print("-" * level + component.get_name() + ": " + component.get_description())
    for child in component.get_children():
        traverse(child, level + 1)

traverse(company)

pidancode.com: 一个神奇的编程公司
-技术部: 负责公司的技术开发
--小明: 技术部的工程师
-人力资源部: 负责公司的招聘和管理
--小红: 人力资源部的招聘专员