Python3如何为IntEnum子类通用函数加类型注解消除Mypy报错
解决方案
方案1:直接在父类添加END类型标注(最简便)
你只需要给父类_Event补充END的类型注解,不需要实际赋值,这样父类仍然是没有枚举成员的空枚举,不影响子类继承,同时Mypy会识别到所有子类都应该包含该属性:
import enum from typing import ClassVar class _Event(enum.IntEnum): # 仅添加类型标注,不实际赋值 END: ClassVar['_Event'] # 原有通用属性、方法保持不变
修改后原函数的类型注解不需要变动即可消除报错。
方案2:使用Protocol定义结构化类型(最严谨)
如果你不想修改父类定义,可以通过typing.Protocol定义约定的结构,所有包含END属性的_Event子类都会自动匹配该协议,不需要显式继承:
import enum from typing import Protocol, ClassVar class _Event(enum.IntEnum): # 原有通用实现保持不变 pass class EventType(Protocol): # 约定所有符合类型必须包含END类属性 END: ClassVar[_Event] # 如果需要继承_Event的其他方法/属性,可在协议中补充对应标注 # 修改函数参数的类型注解为协议类型 # 如果参数是枚举类本身,注解用type[EventType];如果是枚举实例,注解用EventType def generic_event_func(events: type[EventType]): events.END.action = <expr>
临时兼容方案
如果只是需要快速消除单个位置的报错,可以在报错行末尾加忽略注释(不推荐长期使用,会屏蔽真实的属性不存在错误):
def generic_event_func(events: _Event): events.END.action = <expr> # type: ignore[attr-defined]
额外注意
如果你的函数接收的是枚举类本身而非枚举实例,原参数注解events: _Event是错误的,应该修改为events: type[_Event],前者表示枚举实例的类型,后者才表示枚举类的类型,这也是常见的属性报错诱因。
内容的提问来源于stack exchange,提问作者mikb
相关产品推荐
相关产品推荐

