如何让静态类型检查器强制子类重写库基类的指定方法？

要同时实现强制子类定义属性和强制重写基类特定方法的需求，推荐使用typing_extensions.ABCProtocol，它同时整合了ABC的抽象方法强制重写能力和Protocol的属性/结构检查特性，完美解决你遇到的问题。

具体实现代码 #

from typing_extensions import ABCProtocol, abstractmethod

# 第三方库代码
class BaseFoo:
    def a_really_nice_func(self, intparam: int) -> str:
        print("This is the base behaviour")
        return "base"

# 自定义协议：同时强制属性和方法重写
class OurBaseFooProtocol(ABCProtocol):
    # 强制子类必须拥有description属性（类/实例属性均可）
    description: str
    
    # 强制子类必须重写该方法，无视BaseFoo的已有实现
    @abstractmethod
    def a_really_nice_func(self, intparam: int) -> str:
        ...

# 符合要求的子类实现
class Foo(BaseFoo, OurBaseFooProtocol):
    description = "I like trains"
    
    def a_really_nice_func(self, intparam: int) -> str:
        print(f"Custom behaviour with param: {intparam}")
        return f"custom_{intparam}"

# 错误示例1：缺少description属性，mypy/pyright会直接报错
class BadFoo1(BaseFoo, OurBaseFooProtocol):
    def a_really_nice_func(self, intparam: int) -> str:
        return "bad"

# 错误示例2：未重写a_really_nice_func，类型检查器会报错
class BadFoo2(BaseFoo, OurBaseFooProtocol):
    description = "Bad foo"

方案原理 #

• ABCProtocol继承自ABC和Protocol，兼具两者特性：
  1. 作为ABC，@abstractmethod会强制子类必须重写目标方法，无论父类是否已有实现；
  2. 作为Protocol，会自动检查子类是否存在description属性，类型检查器会在属性缺失时即时报错，解决了纯ABC方案无法检查属性的问题。

备选方案（显式标记重写） #

如果不想引入typing_extensions，可以用@override装饰器配合类型检查器的严格模式，强制开发者显式标记重写行为：

from typing_extensions import override
from typing import Protocol

class BaseFoo:
    def a_really_nice_func(self, intparam: int) -> str:
        print("This is the base behaviour")
        return "base"

class OurBaseFooProtocol(Protocol):
    description: str
    def a_really_nice_func(self, intparam: int) -> str:
        ...

class Foo(BaseFoo, OurBaseFooProtocol):
    description = "I like trains"
    
    @override
    def a_really_nice_func(self, intparam: int) -> str:
        return "custom"

需要在类型检查器中开启严格重写检查：

• mypy添加参数：--strict-override
• pyright配置中设置：strictOverride: true

注意：此方案仅强制显式标记重写，无法阻止子类继承BaseFoo的原有实现，仅作为代码规范约束。

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