Python：基于抽象基类继承实现Reader类自动动态注册的解决方案咨询

解决方案：利用__init_subclass__实现自动注册 #

你想要的「继承抽象基类时自动完成Reader注册」的需求，Python 3.6+提供的__init_subclass__方法正好能完美解决——它允许父类在子类被定义时自动执行逻辑，完全替代手动添加装饰器的重复操作。

下面是修改后的完整代码，我会标注关键改动：

修改后的代码

from abc import ABC, abstractmethod
from pathlib import Path

class DataFrameInterface(object):
    """Basic Interface Class."""
    readers = []
    """List of registered reader objects"""
    def __new__(cls, filepath, rbit="r", **kwargs):
        filepath = Path(filepath)
        # Perform file checks for each registered reader
        for reader in cls.readers:
            if reader._check(filepath):
                return reader(filepath, rbit=rbit, **kwargs)
        # If no suitable reader is implemented, raise exception
        _msg = "The input must be a VTK or HDF data set and not {}.".format(filepath.suffix)
        raise NotImplementedError(_msg)

    # 保留原装饰器（可选，如果你还想支持手动注册特殊Reader的话）
    @classmethod
    def register_reader(cls):
        """ Register reader as subclass of DataFrameInterface.
            For a reader to become available, the class object of the reader must have the active decorator @DataFrameInterface.register_reader(). 
        """
        def decorator(subclass):
            cls.readers.append(subclass)
            return subclass
        return decorator

class BasicReader(ABC):
    """An abstract base class for constructing a common reader interface."""
    
    # ------------------- 关键新增：自动注册逻辑 -------------------
    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        # 当子类继承BasicReader时，自动将子类注册到DataFrameInterface
        DataFrameInterface.readers.append(cls)
    # -------------------------------------------------------------

    def __init__(self, filename, rbit="r", **kwargs):
        """Initialize the reader object."""
        self.rbit = rbit
        self.attrs, self.dataframe = self._load(filename, rbit)
        return

    @abstractmethod
    def get_timeSteps(self):
        pass

    @abstractmethod
    def get_mesh(self):
        pass

    def get_constants(self):
        pass

    @abstractmethod
    def get_data(self):
        pass

    @abstractmethod
    def _check(filepath):
        return False

# 去掉手动装饰器！继承BasicReader会自动完成注册
class HDFReader(BasicReader):
    """A simple reader for hdf datasets."""
    @staticmethod
    def _check(filename):
        return filename.suffix == ".h5"

    @staticmethod
    def _load(filename, rbit):
        # 这里可以填充实际的加载逻辑
        return {}, None

    def get_timeSteps(self, dname="times", dfrom="constants"):
        pass

    def get_constants(self, arr_name, index=None, dtype="index"):
        pass

    def get_data(self, arr_name, timeSteps=None, dtype="point_data"):
        pass

    def get_mesh(self, time=None, mtype="constant"):
        pass

# For testing purpose
if __name__ == "__main__":
    # 测试：此时HDFReader已经被自动注册，能正常匹配.h5文件
    try:
        data = DataFrameInterface("filename.h5")
        print("HDFReader自动注册成功！")
    except Exception as e:
        # 这里因为_load是空实现，会报错，但注册逻辑已经生效
        print(f"注册逻辑正常，仅加载逻辑未实现：{e}")

核心原理说明

1. __init_subclass__的作用：当任何类继承BasicReader时，Python会自动调用BasicReader.__init_subclass__方法，我们在这个方法里执行注册逻辑——把当前子类（比如HDFReader）添加到DataFrameInterface.readers列表中。
2. 完全替代手动装饰器：原来的装饰器本质也是把类添加到readers列表，现在这个逻辑被转移到父类的__init_subclass__中，所有子类继承时自动触发，再也不用手动写@DataFrameInterface.register_reader()了。
3. 兼容原有逻辑：如果你还想保留手动注册的能力（比如某些特殊Reader不需要继承BasicReader），可以保留原装饰器，两者不会冲突。

额外注意事项

• 确保你的Python版本在3.6及以上（__init_subclass__是3.6引入的）。
• BasicReader作为抽象基类，自身不会被注册到readers列表中——因为__init_subclass__只有在子类被创建时才会触发，父类本身的定义不会触发这个方法。

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