Python泛型代码触发AttributeError问题排查求助

尝试理解以下Python泛型代码触发AttributeError的原因，所有方法均为类方法，无需实例化对象，仅需使用类型信息。

代码示例 #

from typing import TypeVar, Generic, Type, ClassVar


T = TypeVar('T', bound='Base')


class Base:

    @classmethod
    def display(cls: Type[T]):
        pass


class A(Base):

    @classmethod
    def display(cls):
        print('A')


class B(Base):

    @classmethod
    def display(cls):
        print('B')


class Class(Generic[T]):
    V = ClassVar[T]

    @classmethod
    def display(cls):
        cls.V.display()


Class[A].display()  # Fails with AttributeError
Class[B].display()

错误信息 #

Traceback (most recent call last):
  File "basic.py", line 36, in <module>
    Class[A].display()
  File "basic.py", line 33, in display
    cls.V.display()
  File "/usr/lib/python3.10/typing.py", line 983, in __getattr__
    return getattr(self.__origin__, attr)
  File "/usr/lib/python3.10/typing.py", line 375, in __getattr__
    raise AttributeError(item)
AttributeError: display

问题原因 #

1. ClassVar仅用于类型检查：Class[V] = ClassVar[T]只是给类型检查器的标注，Python运行时不会自动将泛型参数（如A）绑定到V上，V在运行时没有实际值，只是一个占位符。
2. 泛型构造体不是实际类：Class[A]是泛型类型构造体，不是真正的类实例或类对象。当访问Class[A].V时，拿到的是泛型类的占位属性，而非A类本身，这个占位属性没有display方法，因此触发AttributeError。

解决方案 #

方案1：直接从泛型构造体提取类型参数

修改Class的display方法，通过泛型构造体的__args__属性获取传入的类型参数，无需依赖V：

from typing import TypeVar, Generic, Type

T = TypeVar('T', bound='Base')

class Base:
    @classmethod
    def display(cls: Type[T]):
        pass

class A(Base):
    @classmethod
    def display(cls):
        print('A')

class B(Base):
    @classmethod
    def display(cls):
        print('B')

class Class(Generic[T]):
    @classmethod
    def display(cls):
        # 从泛型构造体的__args__中取出实际类型参数
        type_arg = cls.__args__[0]
        type_arg.display()

Class[A].display()  # 输出A
Class[B].display()  # 输出B

说明：__args__是泛型构造体的内置属性，存储了传入的类型参数元组，直接取第一个元素即可得到A或B类。

方案2：手动绑定V属性

如果需要保留V的设计，可手动给每个泛型构造体的V赋值为对应的类，同时修正V的类型标注（因为T是类类型，需标注为Type[T]）：

from typing import TypeVar, Generic, Type, ClassVar

T = TypeVar('T', bound='Base')

class Base:
    @classmethod
    def display(cls: Type[T]):
        pass

class A(Base):
    @classmethod
    def display(cls):
        print('A')

class B(Base):
    @classmethod
    def display(cls):
        print('B')

class Class(Generic[T]):
    V: ClassVar[Type[T]]  # 修正类型标注为类类型

    @classmethod
    def display(cls):
        cls.V.display()

# 手动绑定V到对应的类
Class[A].V = A
Class[B].V = B

Class[A].display()  # 输出A
Class[B].display()  # 输出B

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