如何创建可添加多方法且保留IDE类型提示的Python类装饰器

我希望通过装饰器实现方法复用，同时保留IDE对新增方法的类型提示功能，示例代码如下：

@methods(methods=[implement_foo, implement_bar])
class K:
    pass

# 或者

@method(methods=[Foo, Bar])
class K:
    pass

k = K()

问题所在

输入k.时，IDE应该识别出.foo()或.bar()方法，但实际无法识别。

我的问题和Stack Overflow上的《How to create a class decorator that can add multiple methods to a class?》类似，但要求仅使用一个装饰器且保留类型提示。

我能实现单个方法的装饰器并正常触发类型提示，但多方法场景无法生效。

单个方法装饰器示例（implement_method）：

def implement_method(cls):
    class Inner(cls):
        def __init__(self, *args, **kargs):
            super(Inner, self).__init__(*args, **kargs)
        def method(self):
            pass
    return Inner

@implement_method
class K:
    pass

该方案下，K类实例的类型提示可正常工作。

我猜测问题出在循环逻辑上，但暂未找到替代方案，以下是我的尝试：

def methods(methods):
    def wrapper(cls):
        for method in methods:
            cls = method(cls)
        return cls
    return wrapper

class Bar:
    def bar(self):
        pass

@methods(methods=[Bar])
class K:
    pass

k = K()
k. # <- 无法识别bar()方法

要实现支持多方法复用且保留类型提示的装饰器，核心是让静态分析工具（IDE）能识别装饰后类的实际结构，以下提供两种可行方案：

方案1：基于类混入的实现 #

通过创建继承原类和所有混入类的新类，让IDE直接识别继承的方法，同时用类型注解明确返回类型：

from typing import TypeVar, Type

T = TypeVar('T')

def methods(*mixins: type) -> Type[T]:
    def decorator(cls: Type[T]) -> Type[T]:
        # 构建新类，继承原类与所有混入类
        class DecoratedClass(cls, *mixins):
            pass
        # 保留原类名，避免调试混淆
        DecoratedClass.__name__ = cls.__name__
        return DecoratedClass
    return decorator

# 定义包含复用方法的混入类
class FooMixin:
    def foo(self) -> None:
        print("foo called")

class BarMixin:
    def bar(self) -> None:
        print("bar called")

# 使用装饰器
@methods(FooMixin, BarMixin)
class K:
    pass

k = K()
k.foo()  # IDE可识别foo方法
k.bar()  # IDE可识别bar方法

这种方案的优势是直观，静态分析工具能直接通过继承关系识别新增方法，完全符合IDE的类型提示逻辑。

方案2：基于函数绑定的实现 #

如果需要通过函数而非类来提供复用方法，可通过setattr将函数绑定为类方法，并手动维护类型注解帮助IDE识别：

from typing import TypeVar, Type, Callable

T = TypeVar('T')

def methods(*funcs: Callable) -> Type[T]:
    def decorator(cls: Type[T]) -> Type[T]:
        for func in funcs:
            # 将函数绑定为类的实例方法
            setattr(cls, func.__name__, func)
            # 手动添加类型注解，确保IDE能识别方法
            if hasattr(cls, '__annotations__'):
                cls.__annotations__[func.__name__] = func
        return cls
    return decorator

# 定义复用方法的函数（第一个参数必须为self）
def implement_foo(self) -> None:
    print("foo called")

def implement_bar(self) -> None:
    print("bar called")

# 使用装饰器
@methods(implement_foo, implement_bar)
class K:
    pass

k = K()
k.foo()  # IDE可识别foo方法
k.bar()  # IDE可识别bar方法

关键注意点 #

• 类混入方案是最推荐的方式，因为完全契合Python的继承机制，静态分析工具无需额外处理就能识别方法。
• 函数绑定方案需要确保函数的第一个参数是self，且手动维护类型注解，否则IDE无法正确识别新增方法。

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