如何编写Python类型提示，校验子类方法是否正确覆写指定父类方法？

问题背景 #

需要为override装饰器编写类型提示，让静态检查工具（如mypy）能够识别子类方法是否正确覆写了指定父类的对应方法——不仅要在运行时抛出异常校验有效性，还要在子类实现与父类不匹配时给出静态警告。

之前尝试的方案存在局限：

• 使用Protocol无法适配可变的函数名
• 直接将父类函数传递给装饰器的方式，导致必须将子类方法的self注解为父类类型，进而引发子类属性访问的类型提示问题

解决方案 #

利用Python的泛型类型提示（TypeVar、ParamSpec）结合Self类型（Python 3.11+，低版本可通过typing_extensions引入），既约束子类方法的签名与父类一致，又保留self的子类类型，解决属性访问的类型问题。

实现方案1：指定父类与方法名 #

先定义必要的泛型和参数规范，再实现装饰器的类型提示与运行时校验逻辑：

from typing import TypeVar, ParamSpec, Callable, Type
from typing_extensions import Self  # Python<3.11需安装typing_extensions

P = ParamSpec('P')
R = TypeVar('R')
ParentT = TypeVar('ParentT')

def override(superclass: Type[ParentT], method_name: str) -> Callable[[Callable[P, R]], Callable[P, R]]:
    def decorator(method: Callable[P, R]) -> Callable[P, R]:
        # 运行时校验：检查父类是否存在目标方法
        if not hasattr(superclass, method_name):
            raise ValueError(f"父类 {superclass.__name__} 中不存在方法 {method_name}")
        
        super_method = getattr(superclass, method_name)
        # 可选：用inspect模块做更严格的签名校验
        import inspect
        if not inspect.signature(method) == inspect.signature(super_method):
            raise TypeError(
                f"方法 {method_name} 签名不匹配：父类签名 {inspect.signature(super_method)}，子类签名 {inspect.signature(method)}"
            )
        
        return method
    return decorator

使用时结合Self类型注解self，确保静态检查工具识别self为子类类型：

class Parent:
    def func(self, x: int) -> str:
        return str(x)

class Child(Parent):
    @override(Parent, "func")
    def func(self: Self, x: int) -> str:
        self.child_attr = 42  # 此处self被识别为Child类型，无类型警告
        return f"Child处理：{x}"

实现方案2：直接传递父类方法 #

通过泛型约束直接关联父类方法与子类方法的签名，同时保留self的子类类型：

from typing import TypeVar, ParamSpec, Callable
from typing_extensions import Self

P = ParamSpec('P')
R = TypeVar('R')
ParentT = TypeVar('ParentT')

def override_method(super_method: Callable[[ParentT, *P], R]) -> Callable[[Callable[[Self, *P], R]], Callable[[Self, *P], R]]:
    def decorator(method: Callable[[Self, *P], R]) -> Callable[[Self, *P], R]:
        # 运行时签名校验
        import inspect
        # 移除self参数后对比签名
        super_sig = inspect.signature(super_method).replace(parameters=list(inspect.signature(super_method).parameters.values())[1:])
        method_sig = inspect.signature(method).replace(parameters=list(inspect.signature(method).parameters.values())[1:])
        if super_sig != method_sig:
            raise TypeError(f"签名不匹配：父类方法签名 {super_sig}，子类方法签名 {method_sig}")
        
        return method
    return decorator

使用示例：

class Parent:
    def func(self, x: int) -> str:
        return str(x)

class Child(Parent):
    @override_method(Parent.func)
    def func(self, x: int) -> str:
        self.child_attr = 42  # self自动识别为Child类型，无类型警告
        return f"Child处理：{x}"

效果说明 #

• 静态检查：mypy等工具会根据父类方法的签名，校验子类方法的参数类型、数量及返回值类型是否一致，不匹配时直接给出警告
• 运行时兜底：保留运行时的签名校验，避免静态检查遗漏的场景出现错误
• 类型兼容：通过Self类型确保子类方法的self始终被识别为子类本身，访问子类属性时不会触发类型提示问题

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