Python3.8中@typing.overload结合Literal与Enum的自动补全异常问题

你的用法完全符合规范，问题出在类型检查工具（如Pyright、Mypy）对Enum成员作为Literal参数的重载匹配支持不完善，而非语法或用法错误。

问题原因 #

规范明确允许Enum值作为typing.Literal的参数，但部分类型检查器早期版本在解析Enum成员的Literal类型时，没有正确区分不同的Enum实例，导致重载匹配时误命中第一个定义的重载项（比如你代码中get(C.B)却匹配到返回A的重载）。而基础类型（如int）的字面量重载匹配逻辑已经非常成熟，所以get_int的表现完全正常。

解决办法 #

方法1：改用Enum成员的value作为Literal参数

直接使用Enum成员的value属性作为Literal参数，让类型检查器以基础类型的方式匹配重载，调用时仍保持使用Enum成员：

import typing
from enum import Enum

class A:
    a: int

class B:
    b: int

class C(Enum):
    A = 1
    B = 2

@typing.overload
def get(v: typing.Literal[C.A.value]) -> A: ...

@typing.overload
def get(v: typing.Literal[C.B.value]) -> B: ...

def get(v) -> typing.Union[A, B]:
    pass

a = get(C.A)  # 自动补全返回A实例，正常
b = get(C.B)  # 自动补全返回B实例，正常

方法2：升级类型检查工具到最新版本

主流类型检查器（如Pyright 1.1.290+、Mypy 1.5+）已经修复了Enum Literal的重载匹配问题，升级后无需修改代码即可正常工作。比如VS Code中使用Pylance的话，可以在设置里检查并更新语言服务版本。

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