如何让__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
相关产品推荐
相关产品推荐

