有一个 Python 泛型类型注释的问题

[bestcondition]

from enum import Enum as _Enum
from typing import TypeVar, Type, Generic, Callable


class BaseModel(_BaseModel):
    pass


T = TypeVar('T')
GT = TypeVar('GT')


# 扩展 enum
class EnumBase(Generic[GT], _Enum):
    # 额外对象
    extra_obj: GT

    def __new__(
            cls: Type[T], value: str, extra_obj: GT = None,
    ) -> T:
        obj = object.__new__(cls)
        obj._value_ = value
        obj.extra_obj = extra_obj
        return obj


class B:
    def __init__(self, value):
        self.value = value


# 这里应该如何注释泛型? 这样写 pycharm 是会提示类型的, 但是会报错
class EnumTest(EnumBase[B]):
    A = 'A', B(1)
    B = 'B', B(2)


foo = EnumTest.A.extra_obj
print(foo.value)

报错内容是

  File "xxx", line 35, in <module>
    class EnumTest(EnumBase[B]):
  File "xxx\enum.py", line 408, in __getitem__
    return cls._member_map_[name]
KeyError: <class '__main__.B'>

也就是说 EnumBase[B]会直接触发 EnumMeta 的__getitem__, 不知道有没有什么好的解决办法

[antness]

```python
class EnumTest(EnumBase):
A: 'EnumBase[B]' = 'A', B(1)
B: 'EnumBase[B]' = 'B', B(2)
```
这样可以实现你的需求吗

[antness]

或者
```python
class EnumTest(EnumBase):
A = 'A', B(1) # type: EnumTest[B]
B = 'B', B(2) # type: EnumTest[B]
```

[NoOneNoBody]

typing.ClassVar

[egen]

感觉这种问题很适合用 gpt 来做分析，用 gpt 几个来回分析后给出了可行的方案

以下是 gpt 的回复：
根据你的代码以及产生的错误，我为您分析了原因并提供了一种解决方案。问题的根源在于 EnumMeta （_Enum 基类的元类）没有为泛型类提供支持。因此，我们可以通过创建一个自定义的元类来解决这个问题。

这里是修改后的代码：
```
from enum import Enum as _Enum, EnumMeta as _EnumMeta
from typing import TypeVar, Type, Generic, Any

T = TypeVar('T')
GT = TypeVar('GT', bound=Type)


class _CustomEnumMeta(_EnumMeta):
def __getitem__(self, item: Any) -> Any:
if not isinstance(item, str): # 当处理泛型时，我们可以直接返回自身
return self
return super().__getitem__(item)


class EnumBase(Generic[GT], _Enum, metaclass=_CustomEnumMeta):
extra_obj: GT

def __new__(
cls: Type[T], value: str, extra_obj: GT = None,
) -> T:
obj = object.__new__(cls)
obj._value_ = value
obj.extra_obj = extra_obj
return obj


class B:
def __init__(self, value):
self.value = value


class EnumTest(EnumBase[B]):
A = 'A', B(1)
B = 'B', B(2)


foo = EnumTest.A.extra_obj
print(foo.value)
```

就是成本高了些，几个回合对话后总成本大概：¥3

[anaf]

啊 这对于从 2.7 开始 3.7 结束的人来说 有些看不懂这个代码了

[ruanimal]

@anaf 确实，而且在动态语言里这么写有点邪道了

[thinszx]

啊...第一次知道原来 python 也有泛型...

[gujigujij]

好像不行. 另外你的扩展 Enum 看着有点奇怪, 如果你想有额外对象, 可以这么写:


```python
class B:
def __init__(self, value):
self.value = value

class EnumBase( _Enum):
def __init__(self, code: str, extra_obj: B = None):
self.code: str = code
self.extra_obj: B = extra_obj




class EnumTest(EnumBase):
A = 'A', B(1)
B = 'B', B(2)


res = EnumTest.A.extra_obj.value
print(res)


```

[gujigujij]

```# markdown test
print("hello world")```

[whitewinds]

# 扩展 enum
class EnumBase(Generic[GT]): # <---
# 额外对象
extra_obj: GT

def __new__(cls: Type[T], value: str, extra_obj: GT = None,) -> T:
obj = object.__new__(cls) # type: Any
obj._value_ = value
obj.extra_obj = extra_obj
return obj


class B:
def __init__(self, value):
self.value = value


class EnumTest(EnumBase[B], _Enum): # <---
A = 'A', B(1)
B = 'B', B(2)


foo = EnumTest.A.extra_obj
print(foo.value)

[bestcondition]

@whitewinds 这个 mixin 写的好！受教了！

[bestcondition]

@egen 可以用，gpt 还真是挺厉害的！不过完全为了泛型注解而改元类的 getitem 感觉有点不妥

[egen]

@bestcondition gpt 只会傻傻的让代码可以通过，但是不一定能用正确的方法解决，用来解决一些问题还勉强可以，用来学习可能会被带歪，哈哈