Python中如何对元类插入方法的类进行MyPy类型检查？

以下方案均满足无需大范围修改class MyClassWithSomeMethod(metaclass=MyMeta): ...格式的要求，且不需要子类继承预定义基类：

方案1：标注元类返回值类型（最优，零改动子类代码） #

只要给自定义元类的__new__方法添加返回值类型标注，告知类型检查器所有由该元类生成的类都实现了对应方法接口即可：

from abc import ABCMeta
from typing import Protocol

# 定义方法接口
class _SupportsSomeMethod(Protocol):
    @classmethod
    def some_method(cls, x: str) -> str: ...

class MyMeta(ABCMeta):
    # 标注__new__返回的类类型都实现了_SupportsSomeMethod接口
    def __new__(mcs, *args, **kwargs) -> type[_SupportsSomeMethod]:
        cls = super().__new__(mcs, *args, **kwargs)
        cls.some_method = classmethod(some_method)
        return cls

改完后你原有子类定义完全不用做任何修改，mypy会自动识别所有用MyMeta作为元类的类都带有some_method方法，不会再报属性不存在的错误。

方案2：类内添加TYPE_CHECKING存根（无需修改元类） #

如果不能修改元类定义，可以在目标类中加入仅对类型检查器可见的方法存根，运行时不会生效，也不会覆盖元类动态添加的方法：

from typing import TYPE_CHECKING

class MyClassWithSomeMethod(metaclass=MyMeta):
    if TYPE_CHECKING:
        # 仅类型检查阶段可见的方法定义
        @classmethod
        def some_method(cls, x: str) -> str: ...

仅需在原有类定义中增加3行代码，完全符合最小改动要求。

方案3：解决Protocol元类冲突（适合多个类复用接口的场景） #

之前使用Protocol报错是因为Protocol默认的元类是ProtocolMeta，和你的MyMeta没有继承关系导致冲突，只需要让你的自定义元类同时继承ProtocolMeta即可解决：

from typing import Protocol, ProtocolMeta
from abc import ABCMeta

class SupportsSomeMethod(Protocol):
    @classmethod
    def some_method(cls, x: str) -> str: ...

# 让自定义元类继承ProtocolMeta，解决多基类元类冲突
class MyMeta(ABCMeta, ProtocolMeta):
    def __new__(mcs, *args, **kwargs):
        cls = super().__new__(mcs, *args, **kwargs)
        cls.some_method = classmethod(some_method)
        return cls

# 原有类定义仅需添加Protocol继承
class MyClassWithSomeMethod(SupportsSomeMethod, metaclass=MyMeta):
    pass

该方案适合有大量类需要复用同一个方法接口的场景，接口定义可以统一维护。

方案4：调用处类型强转（临时解决方案） #

如果既不能改元类也不想改类定义，可以在调用方法的地方用cast做一次类型转换，仅需修改调用逻辑：

from typing import cast, Protocol, Type

class SupportsSomeMethod(Protocol):
    @classmethod
    def some_method(cls, x: str) -> str: ...

def call_some_method(cls: Type[MyClassWithSomeMethod]) -> str:
    # 强转成支持方法的接口类型
    return cast(SupportsSomeMethod, cls).some_method("A")

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