如何搭建支持动态调用子类方法的Python项目结构

现有文件结构合理性说明 #

你的现有项目结构是完全合理的，单类单文件是Python项目组织类的通用规范，不需要调整。

最优实现方案 #

核心思路是通过包的__init__.py自动完成子类的导入与注册，同时提供带类型标注的工厂函数，兼顾扩展性与类型检查/代码检查的兼容性。

步骤1：给myclasses包添加__init__.py #

在myclasses目录下新建__init__.py文件，内容如下：

import importlib
import pkgutil
from pathlib import Path
from typing import Dict, Type
from baseclass import BaseClass

# 带类型标注的子类注册表，类型检查器可识别
RENDERER_CLASSES: Dict[str, Type[BaseClass]] = {}

# 自动遍历包下所有模块，导入并注册子类
for _, module_name, _ in pkgutil.iter_modules([Path(__file__).parent]):
    # 跳过私有模块
    if module_name.startswith("_"):
        continue
    # 导入当前模块
    module = importlib.import_module(f".{module_name}", package=__name__)
    # 筛选模块内的BaseClass子类
    for attr_name in dir(module):
        attr = getattr(module, attr_name)
        if (
            isinstance(attr, type)
            and issubclass(attr, BaseClass)
            and attr is not BaseClass
        ):
            # 可用自定义类属性（如attr.renderer_name）作为key，稳定性更高
            RENDERER_CLASSES[attr.__name__] = attr

# 类型友好的工厂函数
def get_renderer(renderer_name: str, *args, **kwargs) -> BaseClass:
    if renderer_name not in RENDERER_CLASSES:
        raise ValueError(
            f"未知渲染器类型：{renderer_name}\n可选类型：{list(RENDERER_CLASSES.keys())}"
        )
    return RENDERER_CLASSES[renderer_name](*args, **kwargs)

步骤2：主程序调用方式 #

在__main__.py中直接调用即可，不需要逐个导入子类：

from myclasses import get_renderer, RENDERER_CLASSES

# 示例1：批量创建所有子类实例并调用render方法
for renderer_name in RENDERER_CLASSES:
    renderer = get_renderer(renderer_name)
    renderer.render()

# 示例2：单独创建指定子类实例
pdf_renderer = get_renderer("SubClass1")
pdf_renderer.render()

方案优势 #

• 完全符合Pythonic规范，仅依赖Python标准库，无额外依赖
• 扩展性极强：新增子类只需要在myclasses目录下添加对应文件，无需修改其他代码
• 完全兼容代码检查/类型检查工具：所有变量、函数都带明确类型标注，mypy、pylint、pylance等工具不会报错
• 没有冗余代码：不需要编写数十条重复导入语句，也不需要使用易引发问题的通配符导入

可选优化 #

如果需要规避类名变更带来的调用问题，可以给每个子类添加固定标识类属性，比如：

# subclass1.py中
from baseclass import BaseClass

class SubClass1(BaseClass):
    # 固定标识，不会随类名变更变化
    renderer_id = "pdf_exporter"
    
    def render(self):
        # 具体实现
        pass

然后修改__init__.py中的注册逻辑，用attr.renderer_id作为RENDERER_CLASSES的key即可。

内容的提问来源于stack exchange，提问作者Ryan