利用Python做一个公司管理系统

class Employee:
    def __init__(self, name, position, salary):
        self.name = name
        self.position = position
        self.salary = salary

class Department:
    def __init__(self, name, manager):
        self.name = name
        self.manager = manager
        self.employees = []

    def add_employee(self, employee):
        self.employees.append(employee)

    def remove_employee(self, employee):
        self.employees.remove(employee)


class CompanyManagementSystem:
    def __init__(self):
        self.employees = []
        self.departments = []

    def add_employee(self, name, position, salary):
        employee = Employee(name, position, salary)
        self.employees.append(employee)

    def remove_employee(self, employee):
        self.employees.remove(employee)

    def add_department(self, name, manager):
        department = Department(name, manager)
        self.departments .append(department)

    def remove_department(self, department):
        self.departments.remove(department)

    def search_employee(self, name):
        matching_employees = []
        for employee in self.employees:
            if name.lower() in employee.name.lower():
                matching_employees.append(employee)
        return matching_employees

    def search_department(self, name):
        matching_departments = []
        for department in self.departments:
            if name.lower() in department.name.lower():
                matching_departments.append(department)
        return matching_departments

# 示例用法
cms = CompanyManagementSystem()

# 添加员工
cms.add_employee("John Doe", "Manager", 5000)
cms.add_employee("Jane Smith", "Assistant", 3000)

# 添加部门
cms.add_department("Sales", "John Doe")
cms.add_department("Marketing", "Jane Smith")

# 查找员工
matching_employees = cms.search_employee("John")
for employee in matching_employees:
    print(employee.name, employee.position, employee.salary)

# 查找部门
matching_departments = cms.search_department("Sales")
for department in matching_departments:
    print(department.name, department.manager)