You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何让__call__继承通过getattr转发的目标方法的类型提示?

让类的__call__方法转发并显示目标方法的类型提示

要实现__call__方法调用转发至指定方法,同时让静态检查工具(如Pyright/Pylance)识别目标方法的类型提示,且支持子类重写目标方法后自动更新调用逻辑,可采用以下两种方案:

方案1:泛型Protocol绑定签名(推荐)

通过定义泛型Protocol描述目标方法的签名,让类实现该Protocol,并保持__call__的签名与目标方法一致。子类重写目标方法时,只需更新泛型参数,类型提示会自动同步。

from typing import Protocol, TypeVar, Self

# 定义参数和返回值的泛型类型变量
T = TypeVar('T')
R = TypeVar('R')

# 用Protocol规范目标方法的签名
class CallTargetProtocol(Protocol[T, R]):
    def num_to_str(self, num: T) -> R: ...

class MyClass(CallTargetProtocol[int, str]):
    call_func_name = 'num_to_str'

    def num_to_str(self, num: int) -> str:
        return str(num)

    # __call__签名与目标方法完全匹配
    def __call__(self, num: int) -> str:
        return getattr(self, self.call_func_name)(num)

# 子类重写目标方法并更新泛型约束
class MySubClass(CallTargetProtocol[float, str]):
    def num_to_str(self, num: float) -> str:
        return f"Processed float: {num}"

# 测试类型提示
mc = MyClass()
mc(123)  # 此处会提示参数为int,返回值为str

msc = MySubClass()
msc(3.14)  # 此处会提示参数为float,返回值为str

方案2:直接复用目标方法的类型注解

如果不需要严格的Protocol约束,可直接让__call__的类型注解与目标方法保持一致,子类重写时同步更新__call__的注解即可。

from typing import Self

class MyClass:
    call_func_name = 'num_to_str'

    def num_to_str(self, num: int) -> str:
        return str(num)

    # 复用目标方法的类型签名
    def __call__(self, num: int) -> str:
        return getattr(self, self.call_func_name)(num)

class MySubClass(MyClass):
    def num_to_str(self, num: float) -> str:
        return f"Float: {num}"

    # 同步更新__call__的签名以匹配重写后的方法
    def __call__(self, num: float) -> str:
        return super().__call__(num)

核心要点

  • 使用getattr(self, self.call_func_name)而非直接赋值__call__ = num_to_str,确保子类重写目标方法后,__call__会调用最新的实例方法,避免静态绑定导致的逻辑不一致。
  • 方案1的泛型Protocol更适合复杂场景,无需手动同步__call__的注解;方案2更简洁,适合简单场景。

内容的提问来源于stack exchange,提问作者joudan

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.04 22:25:28