Python如何基于构造函数参数重载类方法的返回值类型提示

方案1：泛型类+静态类型推导（无需修改运行逻辑，仅加类型标注） #

这是最符合你要求的方案，仅通过增加类型标注就能让静态类型检查器根据实例化入参自动推导gimmie的返回类型，不会出现联合类型提示：

from typing import Generic, Literal, TypeVar, Tuple

# 绑定布尔字面量的类型变量
_IsReturnTuple = TypeVar('_IsReturnTuple', bound=Literal[True, False])

class MyClass(Generic[_IsReturnTuple]):
    def __init__ (self, param1: Tuple[str,...], param2: _IsReturnTuple) -> None:
        self.member1 = param1
        self.member2 = param2
        self.member3 = 10

    # Python 3.12+ 支持直接用条件类型标注返回值
    def gimmie(self) -> Tuple[str,...] if _IsReturnTuple is Literal[True] else int:
        return self.member1 if self.member2 else self.member3

低版本Python可以用@overload装饰器实现同样的效果，无需修改任何运行时代码。
实例化后类型检查器会自动识别返回类型：

• obj1 = MyClass(("a", "b"), True)：obj1.gimmie()返回类型被识别为Tuple[str,...]
• obj2 = MyClass(("a", "b"), False)：obj2.gimmie()返回类型被识别为int

方案2：直接类型断言（最简操作） #

如果你不需要严格的静态校验，只是想在调用时消除联合类型提示，直接用cast做类型转换即可，仅需一行代码：

from typing import cast
obj = MyClass(("test",), True)
# 手动指定返回类型，类型检查器会直接采用你标注的类型
res = cast(Tuple[str, ...], obj.gimmie())

注意该方式需要你自行保证入参param2和断言类型一致，类型检查器不会做运行时校验。

更合理的架构设计 #

如果后续两类返回值的逻辑会有差异化迭代，当前混合设计确实存在职责不清晰的问题，推荐拆分子类+工厂模式实现，运行时和类型层面都完全隔离两类逻辑：

from abc import ABC, abstractmethod
from typing import Tuple

class MyBaseClass(ABC):
    def __init__(self, param1: Tuple[str, ...]) -> None:
        self.member1 = param1
        self.member3 = 10

    @abstractmethod
    def gimmie(self) -> int | Tuple[str, ...]:
        pass

class TupleReturnClass(MyBaseClass):
    def gimmie(self) -> Tuple[str, ...]:
        return self.member1

class IntReturnClass(MyBaseClass):
    def gimmie(self) -> int:
        return self.member3

# 对外暴露的工厂函数
def create_my_class(param1: Tuple[str, ...], return_tuple: bool) -> TupleReturnClass | IntReturnClass:
    return TupleReturnClass(param1) if return_tuple else IntReturnClass(param1)

该方案的扩展性更强，后续新增返回类型仅需新增子类即可，不需要修改原有逻辑。

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