从子类类方法获取基类泛型类的合理实现方案

我需要实现这样的结构：Bar继承自BaseBar[Foo]，Foo继承自BaseFoo，希望在BaseBar的类方法bar中，自动调用对应Foo子类的类方法foo。

目前用get_args结合索引0的临时方案存在隐患——它完全依赖BaseBar父类的声明顺序（比如BaseBar(Generic[FOO], X, Y)里的第一个父类），如果后续调整父类顺序，这段代码就会失效。

现有代码如下：

from abc import abstractmethod
from typing import TypeVar, Generic, get_args

class X:
    pass

class Y:
    pass

class BaseFoo:

    @classmethod
    @abstractmethod
    def foo(cls):
        raise NotImplementedError


class Foo1(BaseFoo):

    @classmethod    
    def foo(cls):
        print("foo1")


class Foo2(BaseFoo):

    @classmethod
    def foo(cls):
        print("foo2")


FOO = TypeVar('FOO', bound=BaseFoo)


class BaseBar(Generic[FOO], X, Y): # 父类顺序直接影响临时方案的有效性

    @classmethod
    def bar(cls):
        # 依赖索引的临时实现，父类顺序改变就会出错
        foo_cls = get_args(cls.__orig_bases__[0])[0]
        foo_cls.foo()


class Bar1(BaseBar[Foo1]):
    pass


class Bar2(BaseBar[Foo2]):
    pass


Bar1.bar() # 输出foo1
Bar2.bar() # 输出foo2

我们可以通过遍历类的原始基类，精准定位到BaseBar对应的泛型实例，从而提取Foo类型，彻底摆脱对父类顺序的依赖。

修改后的BaseBar类代码如下：

from abc import abstractmethod
from typing import TypeVar, Generic, get_args, get_origin

# ... 其他类定义保持不变

class BaseBar(Generic[FOO], X, Y):

    @classmethod
    def bar(cls):
        # 遍历所有原始基类，找到属于BaseBar的泛型实例
        for base in cls.__orig_bases__:
            if get_origin(base) is BaseBar:
                foo_cls = get_args(base)[0]
                foo_cls.foo()
                break
        else:
            # 处理未找到对应泛型参数的情况
            raise ValueError("当前类未正确继承带泛型参数的BaseBar")

方案优势 #

• 不再依赖父类声明顺序，即使把BaseBar的父类改成BaseBar(X, Y, Generic[FOO])，代码依然能正常运行
• 通过get_origin精准匹配BaseBar的泛型实例，逻辑更清晰，鲁棒性更强

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