如何为Python函数指定Event子类的任意联合类型注解？

要实现这个需求，你可以结合typing.Type和typing.Union（Python 3.10+也可用|语法）定义参数类型，限制传入值必须是Event子类的单个类型或任意数量的联合类型。

具体实现代码 #

from typing import Union, Type

class Event:
    pass

class Event1(Event):
    pass

class Event2(Event):
    pass

class Event3(Event):
    pass

def listen_for_events(event_types: Union[Type[Event], ...]) -> None:
    # 函数逻辑实现
    pass

类型检查效果 #

• 以下调用会通过类型检查：

  listen_for_events(Event1)
  listen_for_events(Event1 | Event3)
  listen_for_events(Event1 | Event2 | Event3)
  

• 以下调用会被类型检查器标记为错误：

  listen_for_events(str | int)
  

原理说明 #

• Type[Event]表示Event类或其子类的类型（即类本身，而非类的实例）。
• Union[Type[Event], ...]表示任意数量Type[Event]的联合类型，既支持单个子类，也支持多个子类通过|组合的联合类型。
• 非Event子类的类型（比如str、int）不属于Type[Event]范畴，会被类型检查器拦截。

如果需要更精确的类型推导，可结合TypeVar绑定Event：

from typing import TypeVar, Type, Union

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

def listen_for_events(event_types: Union[Type[T], ...]) -> None:
    pass

这种写法能让类型检查器更清晰地识别传入的都是Event子类的类型，复杂场景下可提供更准确的类型提示。

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