Python重载中使用Literal字符串实现工厂函数返回类型精准标注

Python 类型系统的重载是按照从上到下的优先级匹配的，你不需要额外实现str AND NOT ...的排除逻辑，只要将更具体的 Literal 重载放在通用重载前面即可满足需求，标准写法如下：

from typing import overload, Literal

# 高优先级：匹配具体clsid字面量的重载
@overload
def get_object(clsid: Literal["Foo"]) -> Foo: ...

@overload
def get_object(clsid: Literal["Bar"]) -> Bar: ...

# 低优先级：匹配所有未命中的字符串入参
@overload
def get_object(clsid: str) -> object: ...

# 具体实现的入参和返回值标注使用最宽泛的兼容类型即可
def get_object(clsid: str) -> object:
    if clsid == "Foo":
        return Foo()
    elif clsid == "Bar":
        return Bar()
    # 剩余业务逻辑

该写法的类型推断效果完全符合预期：

• 调用时传入字面量"Foo"/"Bar"，返回值会自动推断为对应的Foo/Bar类型
• 调用时传入其他字符串字面量、或非静态字面量的str变量时，返回值会推断为object类型

如果你能穷举所有用到的clsid，更严谨的常用写法是直接限制入参的取值范围：

from typing import Literal, overload

# 统一定义所有支持的clsid，新增类型时只需修改此处
SupportedClsID = Literal["Foo", "Bar"]

@overload
def get_object(clsid: Literal["Foo"]) -> Foo: ...
@overload
def get_object(clsid: Literal["Bar"]) -> Bar: ...

# 具体实现直接限制入参为支持的clsid集合
def get_object(clsid: SupportedClsID) -> Foo | Bar:
    if clsid == "Foo":
        return Foo()
    return Bar()

这种写法会在静态检查阶段直接拦截非法clsid的调用，避免运行时报错，维护成本也很低：新增clsid时只需要在SupportedClsID中添加字面量，同时新增一条对应的重载规则即可。

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