如何从类装饰器访问TypeVariable实例？解决Pyright类型匹配报错

问题重现 #

给继承自Base的类编写带参数的装饰器时，代码运行逻辑正常，但Pyright静态检查会在替换cls.__init__时抛出类型不匹配错误：

cannot be assigned to member "init" of class "type"

尝试将装饰器参数类型从Type[T]改为T时，又会出现新的类型错误：

Argument of type "Type[Base]" cannot be assigned to parameter of type "T@Decorator"
  Type "Type[Base]" cannot be assigned to type "Base"
    "Type[type]" is incompatible with "Type[Base]"

核心矛盾在于：类装饰器接收的是类对象（Type[T]），但我们需要在装饰器内操作类的实例方法（__init__），Pyright无法自动推断两者的类型兼容性。

解决方案 #

方法1：使用typing.cast强制类型兼容 #

既然明确代码逻辑正确，只是Pyright的类型推断存在局限，可以用cast告诉类型检查工具，自定义的init函数符合类__init__方法的类型要求。

修改后的代码：

from typing import Any, Callable, TypeVar, Type, cast

class Base:
    def __init__(self, config: str):
        self.config = config    

T = TypeVar('T', bound=Base)

def Decorator(arg: str) -> Callable[[Type[T]], Type[T]]:
    def wrapper(cls: Type[T]) -> Type[T]:
        _init = cls.__init__

        def init(self: T, config: str, *args: Any, **kwargs: Any):
            _init(self, config, *args, **kwargs)  
            setattr(self, arg, config)

        # 用cast强制类型转换，消除Pyright报错
        cls.__init__ = cast(Callable[[T, str, *Any, **Any], None], init)
        return cls
    return wrapper

@Decorator('test')
class Test(Base):
    test: str

    def task(self):
        print("test module dummy task: ", self.test)

方法2：用Protocol定义__init__方法的类型规范 #

通过定义一个协议（Protocol），明确要求类必须具备特定签名的__init__方法，让Pyright能正确识别类型兼容性。

from typing import Any, Callable, TypeVar, Type, Protocol

class Base:
    def __init__(self, config: str):
        self.config = config    

# 定义协议，描述符合要求的__init__方法
class HasConfigInit(Protocol):
    def __init__(self, config: str, *args: Any, **kwargs: Any) -> None: ...

# 绑定TypeVar到Base和HasConfigInit的交集
T = TypeVar('T', bound=Base, HasConfigInit)

def Decorator(arg: str) -> Callable[[Type[T]], Type[T]]:
    def wrapper(cls: Type[T]) -> Type[T]:
        _init = cls.__init__

        def init(self: T, config: str, *args: Any, **kwargs: Any):
            _init(self, config, *args, **kwargs)  
            setattr(self, arg, config)

        cls.__init__ = init
        return cls
    return wrapper

@Decorator('test')
class Test(Base):
    test: str

    def task(self):
        print("test module dummy task: ", self.test)

说明 #

两种方法都能解决Pyright的类型检查报错，方法1更简洁直接，适合快速修复；方法2更规范，能让类型检查工具更清晰地理解代码的类型约束。

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