如何正确键入提示 Python 字典，其键为 Type[T] 且值为 generics 该类型？

Question

我想要一个 dict 提示，以便值包含与键相同的类型的 generics ：

from abc import ABC
from typing import Dict, List, Type, TypeVar

class Event(ABC):
    pass

class MessageReceived(Event):
    pass

class MessageSent(Event):
    pass

EventT = TypeVar("EventT", bound=Event)

events: Dict[Type[EventT], List[EventT]] = {}

mypy 返回如下错误：

Type variable "EventT" is unbound  [valid-type]

我理解为什么EventT ，但我无法找到一种方法来正确地暗示这一点。

Answer 1

考虑声明一个表示事件类型和事件列表之间关系的EventRecord类型。

from abc import ABC
from dataclasses import dataclass, field
from typing import Generic, TypeVar


class Event(ABC):
    pass


class MessageReceived(Event):
    pass


class MessageSent(Event):
    pass


EventT = TypeVar("EventT", bound=Event)


@dataclass
class EventRecord(Generic[EventT]):
    event_type: type[EventT]
    events: list[EventT] = field(default_factory=list)


event_registry = {
    EventRecord(event_type=MessageSent),
    EventRecord(event_type=MessageReceived),
}