如何基于表达式为Python函数设置返回类型提示

嘿，我完全懂你的感受——从C转Python，这种在C里习以为常的decltype用法，刚接触Python类型提示时确实会找不到北。不过咱们有几种办法能实现类似的效果，让类型检查器（比如mypy、pyright）正确识别返回类型，而且支持良好的IDE提示。

方法1：用泛型TypeVar捕获返回类型 #

我们可以额外定义一个TypeVar来专门捕获obj.operation()的返回类型，然后把它用作函数的返回类型标注。配合泛型类的参数，类型检查器会自动完成推导：

from typing import TypeVar, Generic
from dataclasses import dataclass

T = TypeVar("T")
R = TypeVar("R")  # 用来捕获operation方法的返回类型

@dataclass
class Wrapper(Generic[T]):
    obj: T

    def operation(self) -> R:
        # 主流类型检查器会自动推断R就是self.obj.operation()的返回类型
        return self.obj.operation()

这种写法不需要额外的接口定义，简单直接，mypy、pyright这类工具都能完美处理。

方法2：用Protocol定义约束（更严谨） #

如果希望在编译期就确保传入Wrapper的对象一定有operation方法，同时明确其返回类型，可以用Protocol定义一个抽象接口。这样不仅能推导返回类型，还能提前拦截不符合要求的类型：

from typing import TypeVar, Generic, Protocol
from dataclasses import dataclass

R = TypeVar("R", covariant=True)

# 定义一个协议，要求实现类必须有返回R类型的operation方法
class HasOperation(Protocol[R]):
    def operation(self) -> R:
        ...

# 约束T必须是实现了HasOperation[R]的类型
T = TypeVar("T", bound=HasOperation[R])

@dataclass
class Wrapper(Generic[T]):
    obj: T

    def operation(self) -> R:
        return self.obj.operation()

这种方式更严谨，能在你传入错误类型时直接给出类型提示，避免运行时错误。

方法3：Python 3.12+ 直接对应decltype的语法 #

Python 3.12引入了类型表达式特性，直接提供了类似decltype的写法——用type[表达式]来标注返回类型，这应该是最贴合你需求的方案：

from typing import Generic, TypeVar
from dataclasses import dataclass

T = TypeVar("T")

@dataclass
class Wrapper(Generic[T]):
    obj: T

    def operation(self) -> type[self.obj.operation()]:
        return self.obj.operation()

这里的type[self.obj.operation()]和C++里的decltype(obj.operation())几乎是完全对应的，类型检查器会直接基于表达式推导返回类型，写法简洁直观。

小提示 #

• 如果你还在使用Python 3.10及以下版本，没法用type[表达式]，推荐用方法1或方法2。
• 记得把你的类型检查器（mypy、pyright）更新到最新版本，才能完美支持这些特性哦。

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